Simple-Web-Server/client_https.hpp

#ifndef CLIENT_HTTPS_HPP
#define CLIENT_HTTPS_HPP

#include "client_http.hpp"

#ifdef USE_STANDALONE_ASIO
#include <asio/ssl.hpp>
#else
#include <boost/asio/ssl.hpp>
#endif

namespace SimpleWeb {
  typedef asio::ssl::stream<asio::ip::tcp::socket> HTTPS;

  template <>
  class Client<HTTPS> : public ClientBase<HTTPS> {
  public:
    Client(const std::string &server_port_path, bool verify_certificate = true, const std::string &cert_file = std::string(),
           const std::string &private_key_file = std::string(), const std::string &verify_file = std::string())
        : ClientBase<HTTPS>::ClientBase(server_port_path, 443), context(asio::ssl::context::tlsv12) {
      if(cert_file.size() > 0 && private_key_file.size() > 0) {
        context.use_certificate_chain_file(cert_file);
        context.use_private_key_file(private_key_file, asio::ssl::context::pem);
      }

      if(verify_certificate)
        context.set_verify_callback(asio::ssl::rfc2818_verification(host));

      if(verify_file.size() > 0)
        context.load_verify_file(verify_file);
      else
        context.set_default_verify_paths();

      if(verify_file.size() > 0 || verify_certificate)
        context.set_verify_mode(asio::ssl::verify_peer);
      else
        context.set_verify_mode(asio::ssl::verify_none);
    }

  protected:
    asio::ssl::context context;

    std::shared_ptr<Connection> create_connection() override {
      return std::make_shared<Connection>(*io_service, context);
    }

    void connect(const std::shared_ptr<Session> &session) override {
      if(!session->connection->socket->lowest_layer().is_open()) {
        auto resolver = std::make_shared<asio::ip::tcp::resolver>(*io_service);
        resolver->async_resolve(*query, [this, session, resolver](const error_code &ec, asio::ip::tcp::resolver::iterator it) {
          auto lock = session->cancel_handlers_mutex->shared_lock();
          if(*session->cancel_handlers)
            return;
          if(!ec) {
            session->set_timeout(this->config.timeout_connect);
            asio::async_connect(session->connection->socket->lowest_layer(), it, [this, session, resolver](const error_code &ec, asio::ip::tcp::resolver::iterator /*it*/) {
              session->cancel_timeout();
              auto lock = session->cancel_handlers_mutex->shared_lock();
              if(*session->cancel_handlers)
                return;
              if(!ec) {
                asio::ip::tcp::no_delay option(true);
                error_code ec;
                session->connection->socket->lowest_layer().set_option(option, ec);

                if(!this->config.proxy_server.empty()) {
                  auto write_buffer = std::make_shared<asio::streambuf>();
                  std::ostream write_stream(write_buffer.get());
                  auto host_port = this->host + ':' + std::to_string(this->port);
                  write_stream << "CONNECT " + host_port + " HTTP/1.1\r\n"
                               << "Host: " << host_port << "\r\n\r\n";
                  session->set_timeout(this->config.timeout_connect);
                  asio::async_write(session->connection->socket->next_layer(), *write_buffer, [this, session, write_buffer](const error_code &ec, size_t /*bytes_transferred*/) {
                    session->cancel_timeout();
                    auto lock = session->cancel_handlers_mutex->shared_lock();
                    if(*session->cancel_handlers)
                      return;
                    if(!ec) {
                      std::shared_ptr<Response> response(new Response());
                      session->set_timeout(this->config.timeout_connect);
                      asio::async_read_until(session->connection->socket->next_layer(), response->content_buffer, "\r\n\r\n", [this, session, response](const error_code &ec, size_t /*bytes_transferred*/) {
                        session->cancel_timeout();
                        auto lock = session->cancel_handlers_mutex->shared_lock();
                        if(*session->cancel_handlers)
                          return;
                        if(!ec) {
                          response->parse_header();
                          if(response->status_code.empty() || response->status_code.compare(0, 3, "200") != 0) {
                            session->connection->close();
                            session->callback(make_error_code::make_error_code(errc::permission_denied));
                          }
                          else
                            this->handshake(session);
                        }
                        else {
                          session->connection->close();
                          session->callback(ec);
                        }
                      });
                    }
                    else {
                      session->connection->close();
                      session->callback(ec);
                    }
                  });
                }
                else
                  this->handshake(session);
              }
              else {
                session->connection->close();
                session->callback(ec);
              }
            });
          }
          else {
            session->connection->close();
            session->callback(ec);
          }
        });
      }
      else
        write(session);
    }

    void handshake(const std::shared_ptr<Session> &session) {
      session->set_timeout(this->config.timeout_connect);
      session->connection->socket->async_handshake(asio::ssl::stream_base::client, [this, session](const error_code &ec) {
        session->cancel_timeout();
        auto lock = session->cancel_handlers_mutex->shared_lock();
        if(*session->cancel_handlers)
          return;
        if(!ec)
          this->write(session);
        else {
          session->connection->close();
          session->callback(ec);
        }
      });
    }
  };
} // namespace SimpleWeb

#endif /* CLIENT_HTTPS_HPP */