SSHServer.hxx

См. документацию.

 
#pragma once
 
#include "SSHConfig.hxx"
#include "SSHSession.hxx"
#include "../../services/ISubject.hxx"
#include "../../core/include/RepositoryManager.hxx"
#include "../../platform/include/NetworkUtils.hxx"
 
#include <cstring>
#include <memory>
#include <string>
#include <atomic>
#include <thread>
#include <vector>
#include <mutex>
 
namespace svcs::server::ssh {

/**
 * @english
 * @class SSHServer
 * @brief Main SSH server class
 * @details Manages SSH connections and executes repository operations
 *
 * @russian
 * @class SSHServer
 * @brief Основной класс SSH сервера
 * @details Управляет SSH соединениями и выполняет операции с репозиториями
 */
class SSHServer {
public:
    SSHServer(const SSHConfig& config,
              std::shared_ptr<svcs::core::RepositoryManager> repo_manager,
              std::shared_ptr<svcs::services::ISubject> event_bus);

    ~SSHServer();
 
    SSHServer(const SSHServer&) = delete;
    SSHServer& operator=(const SSHServer&) = delete;

    bool start();

    void stop();

    bool isRunning() const { return running_; }


    size_t getActiveConnections() const;

    void addUserKey(const std::string& username,
                   const std::string& public_key);

    void setUserPassword(const std::string& username,
                        const std::string& password);
 
private:
    void run();
    void handleClient(svcs::platform::SocketHandle client_fd);
    bool initializeSSH();
    void cleanup();
    void logEvent(const std::string& message, bool is_error = false) const;
    bool handleSVCSCommand(const std::string& command, ssh_channel channel);
 
    SSHConfig config_;
    std::shared_ptr<svcs::core::RepositoryManager> repo_manager_;
    std::shared_ptr<svcs::services::ISubject> event_bus_;
 
    std::atomic<bool> running_{false};
    std::thread server_thread_;
 
    svcs::platform::SocketHandle server_socket_ = INVALID_SOCKET_HANDLE;
    ssh_bind ssh_bind_ = nullptr;
 
    mutable std::mutex connections_mutex_;
    std::vector<std::thread> client_threads_;
    std::atomic<size_t> active_connections_{0};
};
 
}