Chris@16: //
Chris@16: // detail/win_fd_set_adapter.hpp
Chris@16: // ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
Chris@16: //
Chris@101: // Copyright (c) 2003-2015 Christopher M. Kohlhoff (chris at kohlhoff dot com)
Chris@16: //
Chris@16: // Distributed under the Boost Software License, Version 1.0. (See accompanying
Chris@16: // file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
Chris@16: //
Chris@16: 
Chris@16: #ifndef BOOST_ASIO_DETAIL_WIN_FD_SET_ADAPTER_HPP
Chris@16: #define BOOST_ASIO_DETAIL_WIN_FD_SET_ADAPTER_HPP
Chris@16: 
Chris@16: #if defined(_MSC_VER) && (_MSC_VER >= 1200)
Chris@16: # pragma once
Chris@16: #endif // defined(_MSC_VER) && (_MSC_VER >= 1200)
Chris@16: 
Chris@16: #include <boost/asio/detail/config.hpp>
Chris@16: 
Chris@16: #if defined(BOOST_ASIO_WINDOWS) || defined(__CYGWIN__)
Chris@16: 
Chris@16: #include <boost/asio/detail/noncopyable.hpp>
Chris@101: #include <boost/asio/detail/reactor_op_queue.hpp>
Chris@16: #include <boost/asio/detail/socket_types.hpp>
Chris@16: 
Chris@16: #include <boost/asio/detail/push_options.hpp>
Chris@16: 
Chris@16: namespace boost {
Chris@16: namespace asio {
Chris@16: namespace detail {
Chris@16: 
Chris@16: // Adapts the FD_SET type to meet the Descriptor_Set concept's requirements.
Chris@16: class win_fd_set_adapter : noncopyable
Chris@16: {
Chris@16: public:
Chris@16:   enum { default_fd_set_size = 1024 };
Chris@16: 
Chris@16:   win_fd_set_adapter()
Chris@16:     : capacity_(default_fd_set_size),
Chris@16:       max_descriptor_(invalid_socket)
Chris@16:   {
Chris@16:     fd_set_ = static_cast<win_fd_set*>(::operator new(
Chris@16:           sizeof(win_fd_set) - sizeof(SOCKET)
Chris@16:           + sizeof(SOCKET) * (capacity_)));
Chris@16:     fd_set_->fd_count = 0;
Chris@16:   }
Chris@16: 
Chris@16:   ~win_fd_set_adapter()
Chris@16:   {
Chris@16:     ::operator delete(fd_set_);
Chris@16:   }
Chris@16: 
Chris@16:   void reset()
Chris@16:   {
Chris@16:     fd_set_->fd_count = 0;
Chris@16:     max_descriptor_ = invalid_socket;
Chris@16:   }
Chris@16: 
Chris@16:   bool set(socket_type descriptor)
Chris@16:   {
Chris@16:     for (u_int i = 0; i < fd_set_->fd_count; ++i)
Chris@16:       if (fd_set_->fd_array[i] == descriptor)
Chris@16:         return true;
Chris@16: 
Chris@101:     reserve(fd_set_->fd_count + 1);
Chris@16:     fd_set_->fd_array[fd_set_->fd_count++] = descriptor;
Chris@16:     return true;
Chris@16:   }
Chris@16: 
Chris@101:   void set(reactor_op_queue<socket_type>& operations, op_queue<operation>&)
Chris@101:   {
Chris@101:     reactor_op_queue<socket_type>::iterator i = operations.begin();
Chris@101:     while (i != operations.end())
Chris@101:     {
Chris@101:       reactor_op_queue<socket_type>::iterator op_iter = i++;
Chris@101:       reserve(fd_set_->fd_count + 1);
Chris@101:       fd_set_->fd_array[fd_set_->fd_count++] = op_iter->first;
Chris@101:     }
Chris@101:   }
Chris@101: 
Chris@16:   bool is_set(socket_type descriptor) const
Chris@16:   {
Chris@16:     return !!__WSAFDIsSet(descriptor,
Chris@16:         const_cast<fd_set*>(reinterpret_cast<const fd_set*>(fd_set_)));
Chris@16:   }
Chris@16: 
Chris@16:   operator fd_set*()
Chris@16:   {
Chris@16:     return reinterpret_cast<fd_set*>(fd_set_);
Chris@16:   }
Chris@16: 
Chris@16:   socket_type max_descriptor() const
Chris@16:   {
Chris@16:     return max_descriptor_;
Chris@16:   }
Chris@16: 
Chris@101:   void perform(reactor_op_queue<socket_type>& operations,
Chris@101:       op_queue<operation>& ops) const
Chris@101:   {
Chris@101:     for (u_int i = 0; i < fd_set_->fd_count; ++i)
Chris@101:       operations.perform_operations(fd_set_->fd_array[i], ops);
Chris@101:   }
Chris@101: 
Chris@16: private:
Chris@16:   // This structure is defined to be compatible with the Windows API fd_set
Chris@16:   // structure, but without being dependent on the value of FD_SETSIZE. We use
Chris@16:   // the "struct hack" to allow the number of descriptors to be varied at
Chris@16:   // runtime.
Chris@16:   struct win_fd_set
Chris@16:   {
Chris@16:     u_int fd_count;
Chris@16:     SOCKET fd_array[1];
Chris@16:   };
Chris@16: 
Chris@101:   // Increase the fd_set_ capacity to at least the specified number of elements.
Chris@101:   void reserve(u_int n)
Chris@101:   {
Chris@101:     if (n <= capacity_)
Chris@101:       return;
Chris@101: 
Chris@101:     u_int new_capacity = capacity_ + capacity_ / 2;
Chris@101:     if (new_capacity < n)
Chris@101:       new_capacity = n;
Chris@101: 
Chris@101:     win_fd_set* new_fd_set = static_cast<win_fd_set*>(::operator new(
Chris@101:           sizeof(win_fd_set) - sizeof(SOCKET)
Chris@101:           + sizeof(SOCKET) * (new_capacity)));
Chris@101: 
Chris@101:     new_fd_set->fd_count = fd_set_->fd_count;
Chris@101:     for (u_int i = 0; i < fd_set_->fd_count; ++i)
Chris@101:       new_fd_set->fd_array[i] = fd_set_->fd_array[i];
Chris@101: 
Chris@101:     ::operator delete(fd_set_);
Chris@101:     fd_set_ = new_fd_set;
Chris@101:     capacity_ = new_capacity;
Chris@101:   }
Chris@101: 
Chris@16:   win_fd_set* fd_set_;
Chris@16:   u_int capacity_;
Chris@16:   socket_type max_descriptor_;
Chris@16: };
Chris@16: 
Chris@16: } // namespace detail
Chris@16: } // namespace asio
Chris@16: } // namespace boost
Chris@16: 
Chris@16: #include <boost/asio/detail/pop_options.hpp>
Chris@16: 
Chris@16: #endif // defined(BOOST_ASIO_WINDOWS) || defined(__CYGWIN__)
Chris@16: 
Chris@16: #endif // BOOST_ASIO_DETAIL_WIN_FD_SET_ADAPTER_HPP