Chris@16: //
Chris@16: // detail/impl/socket_select_interrupter.ipp
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_IMPL_SOCKET_SELECT_INTERRUPTER_IPP
Chris@16: #define BOOST_ASIO_DETAIL_IMPL_SOCKET_SELECT_INTERRUPTER_IPP
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_RUNTIME)
Chris@16: 
Chris@16: #if defined(BOOST_ASIO_WINDOWS) \
Chris@16:   || defined(__CYGWIN__) \
Chris@16:   || defined(__SYMBIAN32__)
Chris@16: 
Chris@16: #include <cstdlib>
Chris@16: #include <boost/asio/detail/socket_holder.hpp>
Chris@16: #include <boost/asio/detail/socket_ops.hpp>
Chris@16: #include <boost/asio/detail/socket_select_interrupter.hpp>
Chris@16: #include <boost/asio/detail/throw_error.hpp>
Chris@16: #include <boost/asio/error.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: socket_select_interrupter::socket_select_interrupter()
Chris@16: {
Chris@16:   open_descriptors();
Chris@16: }
Chris@16: 
Chris@16: void socket_select_interrupter::open_descriptors()
Chris@16: {
Chris@16:   boost::system::error_code ec;
Chris@16:   socket_holder acceptor(socket_ops::socket(
Chris@16:         AF_INET, SOCK_STREAM, IPPROTO_TCP, ec));
Chris@16:   if (acceptor.get() == invalid_socket)
Chris@16:     boost::asio::detail::throw_error(ec, "socket_select_interrupter");
Chris@16: 
Chris@16:   int opt = 1;
Chris@16:   socket_ops::state_type acceptor_state = 0;
Chris@16:   socket_ops::setsockopt(acceptor.get(), acceptor_state,
Chris@16:       SOL_SOCKET, SO_REUSEADDR, &opt, sizeof(opt), ec);
Chris@16: 
Chris@16:   using namespace std; // For memset.
Chris@16:   sockaddr_in4_type addr;
Chris@16:   std::size_t addr_len = sizeof(addr);
Chris@16:   memset(&addr, 0, sizeof(addr));
Chris@16:   addr.sin_family = AF_INET;
Chris@101:   addr.sin_addr.s_addr = socket_ops::host_to_network_long(INADDR_LOOPBACK);
Chris@16:   addr.sin_port = 0;
Chris@16:   if (socket_ops::bind(acceptor.get(), (const socket_addr_type*)&addr,
Chris@16:         addr_len, ec) == socket_error_retval)
Chris@16:     boost::asio::detail::throw_error(ec, "socket_select_interrupter");
Chris@16: 
Chris@16:   if (socket_ops::getsockname(acceptor.get(), (socket_addr_type*)&addr,
Chris@16:         &addr_len, ec) == socket_error_retval)
Chris@16:     boost::asio::detail::throw_error(ec, "socket_select_interrupter");
Chris@16: 
Chris@16:   // Some broken firewalls on Windows will intermittently cause getsockname to
Chris@16:   // return 0.0.0.0 when the socket is actually bound to 127.0.0.1. We
Chris@16:   // explicitly specify the target address here to work around this problem.
Chris@101:   addr.sin_addr.s_addr = socket_ops::host_to_network_long(INADDR_LOOPBACK);
Chris@16: 
Chris@16:   if (socket_ops::listen(acceptor.get(),
Chris@16:         SOMAXCONN, ec) == socket_error_retval)
Chris@16:     boost::asio::detail::throw_error(ec, "socket_select_interrupter");
Chris@16: 
Chris@16:   socket_holder client(socket_ops::socket(
Chris@16:         AF_INET, SOCK_STREAM, IPPROTO_TCP, ec));
Chris@16:   if (client.get() == invalid_socket)
Chris@16:     boost::asio::detail::throw_error(ec, "socket_select_interrupter");
Chris@16: 
Chris@16:   if (socket_ops::connect(client.get(), (const socket_addr_type*)&addr,
Chris@16:         addr_len, ec) == socket_error_retval)
Chris@16:     boost::asio::detail::throw_error(ec, "socket_select_interrupter");
Chris@16: 
Chris@16:   socket_holder server(socket_ops::accept(acceptor.get(), 0, 0, ec));
Chris@16:   if (server.get() == invalid_socket)
Chris@16:     boost::asio::detail::throw_error(ec, "socket_select_interrupter");
Chris@16:   
Chris@16:   ioctl_arg_type non_blocking = 1;
Chris@16:   socket_ops::state_type client_state = 0;
Chris@16:   if (socket_ops::ioctl(client.get(), client_state,
Chris@16:         FIONBIO, &non_blocking, ec))
Chris@16:     boost::asio::detail::throw_error(ec, "socket_select_interrupter");
Chris@16: 
Chris@16:   opt = 1;
Chris@16:   socket_ops::setsockopt(client.get(), client_state,
Chris@16:       IPPROTO_TCP, TCP_NODELAY, &opt, sizeof(opt), ec);
Chris@16: 
Chris@16:   non_blocking = 1;
Chris@16:   socket_ops::state_type server_state = 0;
Chris@16:   if (socket_ops::ioctl(server.get(), server_state,
Chris@16:         FIONBIO, &non_blocking, ec))
Chris@16:     boost::asio::detail::throw_error(ec, "socket_select_interrupter");
Chris@16: 
Chris@16:   opt = 1;
Chris@16:   socket_ops::setsockopt(server.get(), server_state,
Chris@16:       IPPROTO_TCP, TCP_NODELAY, &opt, sizeof(opt), ec);
Chris@16: 
Chris@16:   read_descriptor_ = server.release();
Chris@16:   write_descriptor_ = client.release();
Chris@16: }
Chris@16: 
Chris@16: socket_select_interrupter::~socket_select_interrupter()
Chris@16: {
Chris@16:   close_descriptors();
Chris@16: }
Chris@16: 
Chris@16: void socket_select_interrupter::close_descriptors()
Chris@16: {
Chris@16:   boost::system::error_code ec;
Chris@16:   socket_ops::state_type state = socket_ops::internal_non_blocking;
Chris@16:   if (read_descriptor_ != invalid_socket)
Chris@16:     socket_ops::close(read_descriptor_, state, true, ec);
Chris@16:   if (write_descriptor_ != invalid_socket)
Chris@16:     socket_ops::close(write_descriptor_, state, true, ec);
Chris@16: }
Chris@16: 
Chris@16: void socket_select_interrupter::recreate()
Chris@16: {
Chris@16:   close_descriptors();
Chris@16: 
Chris@16:   write_descriptor_ = invalid_socket;
Chris@16:   read_descriptor_ = invalid_socket;
Chris@16: 
Chris@16:   open_descriptors();
Chris@16: }
Chris@16: 
Chris@16: void socket_select_interrupter::interrupt()
Chris@16: {
Chris@16:   char byte = 0;
Chris@16:   socket_ops::buf b;
Chris@16:   socket_ops::init_buf(b, &byte, 1);
Chris@16:   boost::system::error_code ec;
Chris@16:   socket_ops::send(write_descriptor_, &b, 1, 0, ec);
Chris@16: }
Chris@16: 
Chris@16: bool socket_select_interrupter::reset()
Chris@16: {
Chris@16:   char data[1024];
Chris@16:   socket_ops::buf b;
Chris@16:   socket_ops::init_buf(b, data, sizeof(data));
Chris@16:   boost::system::error_code ec;
Chris@16:   int bytes_read = socket_ops::recv(read_descriptor_, &b, 1, 0, ec);
Chris@16:   bool was_interrupted = (bytes_read > 0);
Chris@16:   while (bytes_read == sizeof(data))
Chris@16:     bytes_read = socket_ops::recv(read_descriptor_, &b, 1, 0, ec);
Chris@16:   return was_interrupted;
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)
Chris@16:        // || defined(__CYGWIN__)
Chris@16:        // || defined(__SYMBIAN32__)
Chris@16: 
Chris@16: #endif // !defined(BOOST_ASIO_WINDOWS_RUNTIME)
Chris@16: 
Chris@16: #endif // BOOST_ASIO_DETAIL_IMPL_SOCKET_SELECT_INTERRUPTER_IPP