server_base.hpp

Go to the documentation of this file.

/*
    Copyright (C) 2016 - 2024
    by Sergey Popov <dave@whitevine.net>
    Part of the Battle for Wesnoth Project https://www.wesnoth.org/
 
    This program is free software; you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation; either version 2 of the License, or
    (at your option) any later version.
    This program is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY.
 
    See the COPYING file for more details.
*/
 
/**
 * @file
 * Base class for servers using Wesnoth's WML over TCP protocol.
 */
 
#pragma once
 
#include "exceptions.hpp"
#include "server/common/simple_wml.hpp"
 
#include "utils/variant.hpp"
#include "utils/general.hpp"
#include "utils/optional_fwd.hpp"
 
#ifdef _WIN32
#include "serialization/unicode_cast.hpp"
#endif
 
#include <boost/asio/io_context.hpp>
#include <boost/asio/ip/tcp.hpp>
#ifndef _WIN32
#include <boost/asio/posix/stream_descriptor.hpp>
#endif
#include <boost/asio/signal_set.hpp>
#include <boost/asio/streambuf.hpp>
#include <boost/asio/ssl.hpp>
#include <boost/asio/spawn.hpp>
#include <boost/shared_array.hpp>
 
#include <map>
 
extern bool dump_wml;
 
class config;
 
typedef std::shared_ptr<boost::asio::ip::tcp::socket> socket_ptr;
typedef std::shared_ptr<boost::asio::ssl::stream<socket_ptr::element_type>> tls_socket_ptr;
typedef utils::variant<socket_ptr, tls_socket_ptr> any_socket_ptr;
 
struct server_shutdown : public game::error
{
    boost::system::error_code ec;
    server_shutdown(const std::string& msg, boost::system::error_code ec = {}) : game::error(msg), ec(ec) {}
};
 
/**
 * Base class for implementing servers that use gzipped-WML network protocol
 *
 * The protocol is based on TCP connection between client and server.
 * Before WML payloads can be sent a handshake is required. Handshake process is as follows:
 * - client establishes a TCP connection to server.
 * - client sends 32-bit integer(network byte order) representing protocol version requested.
 * - server receives 32-bit integer. Depending on number received server does the following:
 *   0: unencrypted protocol, send unspecified 32-bit integer for compatibility(current implementation sends 42)
 *   1: depending on whether TLS is enabled on server
 *     if TLS enabled: send 32-bit integer 0 and immediately start TLS, client is expected to start TLS on receiving this 0
 *     if TLS disabled: send 32-bit integer 0xFFFFFFFF, on receiving this client should proceed as with unencrypted connection or immediately close
 *   any other number: server closes connection immediately
 * - at this point handshake is completed and client and server can exchange WML messages
 *
 * Message format is as follows:
 * - 32-bit unsigned integer(network byte order), this is size of the following payload
 * - payload: gzipped WML data, which is WML text fed through gzip utility or the equivalent library function.
 */
class server_base
{
    template<class SocketPtr> void send_doc_queued(SocketPtr socket, std::unique_ptr<simple_wml::document>& doc_ptr, boost::asio::yield_context yield);
 
public:
    server_base(unsigned short port, bool keep_alive);
    virtual ~server_base() {}
    int run();
 
    /**
     * Send a WML document from within a coroutine
     * @param socket
     * @param doc
     * @param yield The function will suspend on write operation using this yield context
     */
    template<class SocketPtr> void coro_send_doc(SocketPtr socket, simple_wml::document& doc, const boost::asio::yield_context& yield);
    /**
     * Send contents of entire file directly to socket from within a coroutine
     * @param socket
     * @param filename
     * @param yield The function will suspend on write operations using this yield context
     */
    void coro_send_file(const socket_ptr& socket, const std::string& filename, const boost::asio::yield_context& yield);
    void coro_send_file(tls_socket_ptr socket, const std::string& filename, const boost::asio::yield_context& yield);
    /**
     * Receive WML document from a coroutine
     * @param socket
     * @param yield The function will suspend on read operation using this yield context
     * @return unique_ptr with doc deceived. In case of error empty unique_ptr
     */
    template<class SocketPtr> std::unique_ptr<simple_wml::document> coro_receive_doc(SocketPtr socket, const boost::asio::yield_context& yield);
 
    /**
     * High level wrapper for sending a WML document
     *
     * This function returns before send is finished. This function can be called again on same socket before previous send was finished.
     * WML documents are kept in internal queue and sent in FIFO order.
     * @param socket
     * @param doc Document to send. A copy of it will be made so there is no need to keep the reference live after the function returns.
     */
    template<class SocketPtr> void async_send_doc_queued(SocketPtr socket, simple_wml::document& doc);
 
    typedef std::map<std::string, std::string> info_table;
    template<class SocketPtr> void async_send_error(SocketPtr socket, const std::string& msg, const char* error_code = "", const info_table& info = {});
    template<class SocketPtr> void async_send_warning(SocketPtr socket, const std::string& msg, const char* warning_code = "", const info_table& info = {});
 
    /**
     * Handles hashing the password provided by the player before comparing it to the hashed password in the forum database.
     *
     * @param pw The plaintext password.
     * @param salt The salt as retrieved from the forum database.
     * @param username The player attempting to log in.
     * @return The hashed password, or empty if the password couldn't be hashed.
     */
    std::string hash_password(const std::string& pw, const std::string& salt, const std::string& username);
 
protected:
    unsigned short port_;
    bool keep_alive_;
    boost::asio::io_context io_service_;
    boost::asio::ssl::context tls_context_ { boost::asio::ssl::context::sslv23 };
    bool tls_enabled_ { false };
    boost::asio::ip::tcp::acceptor acceptor_v6_;
    boost::asio::ip::tcp::acceptor acceptor_v4_;
 
    void load_tls_config(const config& cfg);
 
    void start_server();
    void serve(const boost::asio::yield_context& yield, boost::asio::ip::tcp::acceptor& acceptor, const boost::asio::ip::tcp::endpoint& endpoint);
 
    uint32_t handshake_response_;
 
    virtual void handle_new_client(socket_ptr socket) = 0;
    virtual void handle_new_client(tls_socket_ptr socket) = 0;
 
    virtual bool accepting_connections() const { return true; }
    virtual bool ip_exceeds_connection_limit(const std::string&) const { return false; }
 
    struct login_ban_info
    {
        const char* error_code;
        std::string reason;
        utils::optional<std::chrono::seconds> time_remaining;
    };
 
    virtual utils::optional<login_ban_info> is_ip_banned(const std::string&) { return {}; }
 
#ifndef _WIN32
    boost::asio::posix::stream_descriptor input_;
    std::string fifo_path_;
    void read_from_fifo();
    virtual void handle_read_from_fifo(const boost::system::error_code& error, std::size_t bytes_transferred) = 0;
    boost::asio::streambuf admin_cmd_;
 
    boost::asio::signal_set sighup_;
    virtual void handle_sighup(const boost::system::error_code& error, int signal_number) = 0;
#endif
};
 
template<class SocketPtr> std::string client_address(SocketPtr socket);
template<class SocketPtr> std::string log_address(SocketPtr socket) { return (utils::decayed_is_same<tls_socket_ptr, decltype(socket)> ? "+" : "") + client_address(socket); }
template<class SocketPtr> bool check_error(const boost::system::error_code& error, SocketPtr socket);