ceph/src/boost/libs/asio/example/cpp03/local/stream_server.cpp

   1 //
   2 // stream_server.cpp
   3 // ~~~~~~~~~~~~~~~~~
   4 //
   5 // Copyright (c) 2003-2016 Christopher M. Kohlhoff (chris at kohlhoff dot com)
   6 //
   7 // Distributed under the Boost Software License, Version 1.0. (See accompanying
   8 // file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
   9 //
  10
  11 #include <cstdio>
  12 #include <iostream>
  13 #include <boost/array.hpp>
  14 #include <boost/bind.hpp>
  15 #include <boost/enable_shared_from_this.hpp>
  16 #include <boost/shared_ptr.hpp>
  17 #include <boost/asio.hpp>
  18
  19 #if defined(BOOST_ASIO_HAS_LOCAL_SOCKETS)
  20
  21 using boost::asio::local::stream_protocol;
  22
  23 class session
  24   : public boost::enable_shared_from_this<session>
  25 {
  26 public:
  27   session(boost::asio::io_service& io_service)
  28     : socket_(io_service)
  29   {
  30   }
  31
  32   stream_protocol::socket& socket()
  33   {
  34     return socket_;
  35   }
  36
  37   void start()
  38   {
  39     socket_.async_read_some(boost::asio::buffer(data_),
  40         boost::bind(&session::handle_read,
  41           shared_from_this(),
  42           boost::asio::placeholders::error,
  43           boost::asio::placeholders::bytes_transferred));
  44   }
  45
  46   void handle_read(const boost::system::error_code& error,
  47       size_t bytes_transferred)
  48   {
  49     if (!error)
  50     {
  51       boost::asio::async_write(socket_,
  52           boost::asio::buffer(data_, bytes_transferred),
  53           boost::bind(&session::handle_write,
  54             shared_from_this(),
  55             boost::asio::placeholders::error));
  56     }
  57   }
  58
  59   void handle_write(const boost::system::error_code& error)
  60   {
  61     if (!error)
  62     {
  63       socket_.async_read_some(boost::asio::buffer(data_),
  64           boost::bind(&session::handle_read,
  65             shared_from_this(),
  66             boost::asio::placeholders::error,
  67             boost::asio::placeholders::bytes_transferred));
  68     }
  69   }
  70
  71 private:
  72   // The socket used to communicate with the client.
  73   stream_protocol::socket socket_;
  74
  75   // Buffer used to store data received from the client.
  76   boost::array<char, 1024> data_;
  77 };
  78
  79 typedef boost::shared_ptr<session> session_ptr;
  80
  81 class server
  82 {
  83 public:
  84   server(boost::asio::io_service& io_service, const std::string& file)
  85     : io_service_(io_service),
  86       acceptor_(io_service, stream_protocol::endpoint(file))
  87   {
  88     session_ptr new_session(new session(io_service_));
  89     acceptor_.async_accept(new_session->socket(),
  90         boost::bind(&server::handle_accept, this, new_session,
  91           boost::asio::placeholders::error));
  92   }
  93
  94   void handle_accept(session_ptr new_session,
  95       const boost::system::error_code& error)
  96   {
  97     if (!error)
  98     {
  99       new_session->start();
 100     }
 101
 102     new_session.reset(new session(io_service_));
 103     acceptor_.async_accept(new_session->socket(),
 104         boost::bind(&server::handle_accept, this, new_session,
 105           boost::asio::placeholders::error));
 106   }
 107
 108 private:
 109   boost::asio::io_service& io_service_;
 110   stream_protocol::acceptor acceptor_;
 111 };
 112
 113 int main(int argc, char* argv[])
 114 {
 115   try
 116   {
 117     if (argc != 2)
 118     {
 119       std::cerr << "Usage: stream_server <file>\n";
 120       std::cerr << "*** WARNING: existing file is removed ***\n";
 121       return 1;
 122     }
 123
 124     boost::asio::io_service io_service;
 125
 126     std::remove(argv[1]);
 127     server s(io_service, argv[1]);
 128
 129     io_service.run();
 130   }
 131   catch (std::exception& e)
 132   {
 133     std::cerr << "Exception: " << e.what() << "\n";
 134   }
 135
 136   return 0;
 137 }
 138
 139 #else // defined(BOOST_ASIO_HAS_LOCAL_SOCKETS)
 140 # error Local sockets not available on this platform.
 141 #endif // defined(BOOST_ASIO_HAS_LOCAL_SOCKETS)