ChannelSSHTunnel.h

// Author: Kang Lin <kl222@126.com>
 
#ifndef CCHANNELSSHTUNNEL_H
#define CCHANNELSSHTUNNEL_H
 
#pragma once
 
#include <QSharedPointer>
#include <QSocketNotifier>
#include <QMutex>
 
#include "libssh/libssh.h"
#include "libssh/callbacks.h"
 
#include "Channel.h"
#include "ParameterChannelSSH.h"
#include "Event.h"
 
class CHANNEL_EXPORT CChannelSSHTunnel : public CChannel
{
    Q_OBJECT
 
public:
    explicit CChannelSSHTunnel(QSharedPointer<CParameterChannelSSH> parameter,
                               bool bWakeUp = true,
                               QObject *parent = nullptr);
    virtual ~CChannelSSHTunnel();
 
    // QIODevice interface
public:
    virtual bool open(OpenMode mode) override;
    virtual void close() override;
    
    virtual QString GetDetails() override;
    int Process();
    virtual int WakeUp();
 
Q_SIGNALS:
    void sigBlockShowMessageBox(const QString& szTitle,
                                const QString& szMessage,
                                QMessageBox::StandardButtons buttons,
                                QMessageBox::StandardButton& nRet,
                                bool &checkBox,
                                QString checkBoxContext = QString());
    void sigBlockShowWidget(const QString& className, int &nRet, void* pContext);
 
    // QIODevice interface
protected:
    virtual qint64 readData(char *data, qint64 maxlen) override;
    virtual qint64 writeData(const char *data, qint64 len) override;
 
private:
    int verifyKnownhost(ssh_session session);
    int authentication(
        ssh_session session,
        const QString szUser,
        const QString szPassword,
        const QString szPassphrase,
        const int nMethod = SSH_AUTH_METHOD_PASSWORD);
    int authenticationPublicKey(
        ssh_session session,
        const QString szUser,
        const QString szPublicKeyFile,
        const QString szPrivateKeyFile,
        const QString szPassphrase);
    int forward(ssh_session session);
 
    static void cb_log(ssh_session session,
                       int priority,
                       const char *message,
                       void *userdata);
 
protected:
    ssh_session m_Session;
    ssh_channel m_Channel;
 
private:
    ssh_pcap_file m_pcapFile;
    QSharedPointer<CParameterChannelSSH> m_Parameter;
    
    QSocketNotifier* m_pSocketRead;
    QSocketNotifier* m_pSocketWrite;
    QSocketNotifier* m_pSocketException;
    
    Channel::CEvent* m_pEvent;
};
 
#endif // CCHANNELSSHTUNNEL_H