websocket_handshake.h

Go to the documentation of this file.

// BSD 3-Clause License
// Copyright (c) 2024, 🍀☀🌕🌥 🌊
// See the LICENSE file in the project root for full license information.
 
#pragma once
 
#include <cstdint>
#include <map>
#include <string>
#include <string_view>
#include <vector>
 
namespace kcenon::network::internal
{
    struct ws_handshake_result
    {
        bool success;                                  
        std::string error_message;                     
        std::map<std::string, std::string> headers;    
    };
 
    class websocket_handshake
    {
    public:
        static auto create_client_handshake(
            std::string_view host, std::string_view path, uint16_t port,
            const std::map<std::string, std::string>& extra_headers = {})
            -> std::string;
 
        static auto validate_server_response(const std::string& response,
                                              const std::string& expected_key)
            -> ws_handshake_result;
 
        static auto parse_client_request(const std::string& request)
            -> ws_handshake_result;
 
        static auto create_server_response(const std::string& client_key)
            -> std::string;
 
        static auto generate_websocket_key() -> std::string;
 
        static auto calculate_accept_key(const std::string& client_key)
            -> std::string;
 
    private:
        static auto parse_headers(const std::string& http_message)
            -> std::map<std::string, std::string>;
 
        static auto extract_status_code(const std::string& response) -> int;
 
        static auto base64_encode(const std::vector<uint8_t>& data)
            -> std::string;
 
        static auto sha1_hash(const std::string& data) -> std::vector<uint8_t>;
    };
 
} // namespace kcenon::network::internal