TLA  Line data    Source code

       1                 : //
       2                 : // Copyright (c) 2026 Steve Gerbino
       3                 : //
       4                 : // Distributed under the Boost Software License, Version 1.0. (See accompanying
       5                 : // file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
       6                 : //
       7                 : // Official repository: https://github.com/cppalliance/corosio
       8                 : //
       9                 : 
      10                 : #ifndef BOOST_COROSIO_NATIVE_DETAIL_SELECT_SELECT_SOCKET_HPP
      11                 : #define BOOST_COROSIO_NATIVE_DETAIL_SELECT_SELECT_SOCKET_HPP
      12                 : 
      13                 : #include <boost/corosio/detail/platform.hpp>
      14                 : 
      15                 : #if BOOST_COROSIO_HAS_SELECT
      16                 : 
      17                 : #include <boost/corosio/tcp_socket.hpp>
      18                 : #include <boost/capy/ex/executor_ref.hpp>
      19                 : #include <boost/corosio/detail/intrusive.hpp>
      20                 : 
      21                 : #include <boost/corosio/native/detail/select/select_op.hpp>
      22                 : 
      23                 : #include <memory>
      24                 : 
      25                 : namespace boost::corosio::detail {
      26                 : 
      27                 : class select_socket_service;
      28                 : 
      29                 : /// Socket implementation for select backend.
      30                 : class select_socket final
      31                 :     : public tcp_socket::implementation
      32                 :     , public std::enable_shared_from_this<select_socket>
      33                 :     , public intrusive_list<select_socket>::node
      34                 : {
      35                 :     friend class select_socket_service;
      36                 : 
      37                 : public:
      38                 :     explicit select_socket(select_socket_service& svc) noexcept;
      39                 : 
      40                 :     std::coroutine_handle<> connect(
      41                 :         std::coroutine_handle<>,
      42                 :         capy::executor_ref,
      43                 :         endpoint,
      44                 :         std::stop_token,
      45                 :         std::error_code*) override;
      46                 : 
      47                 :     std::coroutine_handle<> read_some(
      48                 :         std::coroutine_handle<>,
      49                 :         capy::executor_ref,
      50                 :         io_buffer_param,
      51                 :         std::stop_token,
      52                 :         std::error_code*,
      53                 :         std::size_t*) override;
      54                 : 
      55                 :     std::coroutine_handle<> write_some(
      56                 :         std::coroutine_handle<>,
      57                 :         capy::executor_ref,
      58                 :         io_buffer_param,
      59                 :         std::stop_token,
      60                 :         std::error_code*,
      61                 :         std::size_t*) override;
      62                 : 
      63                 :     std::error_code shutdown(tcp_socket::shutdown_type what) noexcept override;
      64                 : 
      65 HIT       18799 :     native_handle_type native_handle() const noexcept override
      66                 :     {
      67           18799 :         return fd_;
      68                 :     }
      69                 : 
      70                 :     std::error_code set_option(
      71                 :         int level, int optname,
      72                 :         void const* data, std::size_t size) noexcept override;
      73                 :     std::error_code get_option(
      74                 :         int level, int optname,
      75                 :         void* data, std::size_t* size) const noexcept override;
      76                 : 
      77              19 :     endpoint local_endpoint() const noexcept override
      78                 :     {
      79              19 :         return local_endpoint_;
      80                 :     }
      81              21 :     endpoint remote_endpoint() const noexcept override
      82                 :     {
      83              21 :         return remote_endpoint_;
      84                 :     }
      85                 :     bool is_open() const noexcept
      86                 :     {
      87                 :         return fd_ >= 0;
      88                 :     }
      89                 :     void cancel() noexcept override;
      90                 :     void cancel_single_op(select_op& op) noexcept;
      91                 :     void close_socket() noexcept;
      92            3071 :     void set_socket(int fd) noexcept
      93                 :     {
      94            3071 :         fd_ = fd;
      95            3071 :     }
      96            6142 :     void set_endpoints(endpoint local, endpoint remote) noexcept
      97                 :     {
      98            6142 :         local_endpoint_  = local;
      99            6142 :         remote_endpoint_ = remote;
     100            6142 :     }
     101                 : 
     102                 :     select_connect_op conn_;
     103                 :     select_read_op rd_;
     104                 :     select_write_op wr_;
     105                 : 
     106                 : private:
     107                 :     select_socket_service& svc_;
     108                 :     int fd_ = -1;
     109                 :     endpoint local_endpoint_;
     110                 :     endpoint remote_endpoint_;
     111                 : };
     112                 : 
     113                 : } // namespace boost::corosio::detail
     114                 : 
     115                 : #endif // BOOST_COROSIO_HAS_SELECT
     116                 : 
     117                 : #endif // BOOST_COROSIO_NATIVE_DETAIL_SELECT_SELECT_SOCKET_HPP