ceph/src/boost/libs/asio/example/cpp11/ssl/server.cpp

   1 //
   2 // server.cpp
   3 // ~~~~~~~~~~
   4 //
   5 // Copyright (c) 2003-2020 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 <cstdlib>
  12 #include <functional>
  13 #include <iostream>
  14 #include <boost/asio.hpp>
  15 #include <boost/asio/ssl.hpp>
  16
  17 using boost::asio::ip::tcp;
  18
  19 class session : public std::enable_shared_from_this<session>
  20 {
  21 public:
  22   session(boost::asio::ssl::stream<tcp::socket> socket)
  23     : socket_(std::move(socket))
  24   {
  25   }
  26
  27   void start()
  28   {
  29     do_handshake();
  30   }
  31
  32 private:
  33   void do_handshake()
  34   {
  35     auto self(shared_from_this());
  36     socket_.async_handshake(boost::asio::ssl::stream_base::server,
  37         [this, self](const boost::system::error_code& error)
  38         {
  39           if (!error)
  40           {
  41             do_read();
  42           }
  43         });
  44   }
  45
  46   void do_read()
  47   {
  48     auto self(shared_from_this());
  49     socket_.async_read_some(boost::asio::buffer(data_),
  50         [this, self](const boost::system::error_code& ec, std::size_t length)
  51         {
  52           if (!ec)
  53           {
  54             do_write(length);
  55           }
  56         });
  57   }
  58
  59   void do_write(std::size_t length)
  60   {
  61     auto self(shared_from_this());
  62     boost::asio::async_write(socket_, boost::asio::buffer(data_, length),
  63         [this, self](const boost::system::error_code& ec,
  64           std::size_t /*length*/)
  65         {
  66           if (!ec)
  67           {
  68             do_read();
  69           }
  70         });
  71   }
  72
  73   boost::asio::ssl::stream<tcp::socket> socket_;
  74   char data_[1024];
  75 };
  76
  77 class server
  78 {
  79 public:
  80   server(boost::asio::io_context& io_context, unsigned short port)
  81     : acceptor_(io_context, tcp::endpoint(tcp::v4(), port)),
  82       context_(boost::asio::ssl::context::sslv23)
  83   {
  84     context_.set_options(
  85         boost::asio::ssl::context::default_workarounds
  86         | boost::asio::ssl::context::no_sslv2
  87         | boost::asio::ssl::context::single_dh_use);
  88     context_.set_password_callback(std::bind(&server::get_password, this));
  89     context_.use_certificate_chain_file("server.pem");
  90     context_.use_private_key_file("server.pem", boost::asio::ssl::context::pem);
  91     context_.use_tmp_dh_file("dh2048.pem");
  92
  93     do_accept();
  94   }
  95
  96 private:
  97   std::string get_password() const
  98   {
  99     return "test";
 100   }
 101
 102   void do_accept()
 103   {
 104     acceptor_.async_accept(
 105         [this](const boost::system::error_code& error, tcp::socket socket)
 106         {
 107           if (!error)
 108           {
 109             std::make_shared<session>(
 110                 boost::asio::ssl::stream<tcp::socket>(
 111                   std::move(socket), context_))->start();
 112           }
 113
 114           do_accept();
 115         });
 116   }
 117
 118   tcp::acceptor acceptor_;
 119   boost::asio::ssl::context context_;
 120 };
 121
 122 int main(int argc, char* argv[])
 123 {
 124   try
 125   {
 126     if (argc != 2)
 127     {
 128       std::cerr << "Usage: server <port>\n";
 129       return 1;
 130     }
 131
 132     boost::asio::io_context io_context;
 133
 134     using namespace std; // For atoi.
 135     server s(io_context, atoi(argv[1]));
 136
 137     io_context.run();
 138   }
 139   catch (std::exception& e)
 140   {
 141     std::cerr << "Exception: " << e.what() << "\n";
 142   }
 143
 144   return 0;
 145 }