SSHSession.hxx

Go to the documentation of this file.

#pragma once
 
#if defined(__GNUC__)
#pragma GCC diagnostic push
#pragma GCC diagnostic ignored "-Wdeprecated-declarations"
#endif
 
#include "SSHConfig.hxx"
#include "../../platform/include/NetworkUtils.hxx"
#include <libssh/libssh.h>
#include <libssh/server.h>
#include <memory>
#include <string>
#include <functional>
#include <chrono>
 
namespace svcs::server::ssh {

class SSHSession {
public:
    using CommandHandler = std::function<bool(const std::string&, ssh_channel)>;

    SSHSession(ssh_session session,
               const SSHConfig& config,
               CommandHandler handler);

    ~SSHSession();
 
    // Запрещаем копирование
    SSHSession(const SSHSession&) = delete;
    SSHSession& operator=(const SSHSession&) = delete;
 
    // Разрешаем перемещение
    SSHSession(SSHSession&& other) noexcept;
    SSHSession& operator=(SSHSession&& other) noexcept;

    bool process();

    bool isAlive() const;

    std::string getUsername() const { return username_; }

    std::string getClientIp() const;

    std::chrono::system_clock::time_point getStartTime() const { return start_time_; }

    void close();
 
private:
    bool authenticate();
    bool handleCommand();
    bool validatePublicKey(ssh_key key);
    bool validatePassword(const std::string& username, const std::string& password);
    std::vector<std::string> loadAuthorizedKeys(const std::string& username) const;
    std::string getPasswordHash(const std::string& username) const;
 
private:
    ssh_session session_;                 
    SSHConfig config_;                    
    CommandHandler handler_;              
    std::string username_;                
    bool authenticated_ = false;          
    std::chrono::system_clock::time_point start_time_; 
    bool owns_session_ = true;            
};
 
}
 
#if defined(__GNUC__)
#pragma GCC diagnostic pop
#endif