BackendSftpServer.h

// Author: Kang Lin <kl222@126.com>
 
#pragma once
 
#include <QSocketNotifier>
 
#include <libssh/server.h>
#include <libssh/sftpserver.h>
#include <libssh/callbacks.h>
 
#include "BackendServer.h"
#include "OperateSftpServer.h"
#include "ParameterSftpServer.h"

class CBackendSftpServer : public CBackendServer
{
    Q_OBJECT
 
public:
    explicit CBackendSftpServer(COperateSftpServer *pOperate = nullptr,
                                bool bStopSignal = true);
    virtual ~CBackendSftpServer();
 
public Q_SLOTS:
    virtual void slotDisconnect(const QString& szIp, const quint16 port) override;
 
protected:
    virtual OnInitReturnValue OnInit() override;
    virtual int OnClean() override;
    virtual int OnProcess() override;
    virtual int WakeUp() override;
 
private Q_SLOTS:
    void slotNewConnection();
 
private:
    CParameterSftpServer* m_pPara;
    ssh_event m_event;
 
    struct sBindData {
        ssh_bind sshBind;
        QSocketNotifier* pListenNotifier;
    };
    QList<sBindData*> m_lstListen;
    struct sClientData
    {
        QString ip;
        quint16 port;
        ssh_session session;
        ssh_channel channel;
        sftp_session sftp;
        QTime m_Time;
        int m_nAuthAttempts;
        struct ssh_channel_callbacks_struct channel_cb;
        struct ssh_server_callbacks_struct server_cb;
        CParameterSftpServer* pPara;
    };
    QList<sClientData*> m_lstClients;
 
    static int cbAuthPassword(ssh_session session, const char *user,
                              const char *pass, void *userdata);
    static int cbAuthPublickey(ssh_session session,
                               const char *user,
                               struct ssh_key_struct *pubkey,
                               char signature_state,
                               void *userdata);
    static ssh_channel cbChannelOpen(ssh_session session, void *userdata);
 
    int Listen(const QString &szIp, qint16 nPort);
    ssh_bind GetSshBind(QSocketNotifier* notifier);
    bool InitHostKey(ssh_bind ssdBind);
    int DeleteClient(sClientData* pClient);
    static void SendBanner(sClientData* pClient, const QString& user);
};