File indexing completed on 2024-05-12 09:01:21

 /*
  * SPDX-FileCopyrightText: 2023 Arjen Hiemstra <ahiemstra@heimr.nl>
  * SPDX-License-Identifier: GPL-2.0-or-later
  */
 
 #pragma once
 
 #include <atomic>
 #include <memory>
 #include <thread>
 
 #include <QImage>
 #include <QObject>
 #include <QSize>
 
 #include <freerdp/freerdp.h>
 
 class RdpSession;
 class RdpView;
 class RdpHostPreferences;
 
 struct RdpContext {
     rdpContext _c;
 
     RdpSession *session = nullptr;
 };
 
 struct Certificate {
     QString toString() const;
 
     QString host;
     qint16 port;
     QString commonName;
     QString subject;
     QString issuer;
     QString fingerprint;
     int flags;
 };
 
 class RdpSession : public QObject
 {
     Q_OBJECT
 
 public:
     /**
      * Session state.
      */
     enum class State {
         Initial,
         Starting,
         Connected,
         Running,
         Closed,
     };
 
     enum class CertificateResult {
         DoNotAccept,
         AcceptTemporarily,
         AcceptPermanently,
     };
 
     RdpSession(RdpView *view);
     ~RdpSession() override;
 
     /**
      * The current session state.
      */
     State state() const;
     Q_SIGNAL void stateChanged();
 
     QString host() const;
     void setHost(const QString &newHost);
 
     QString user() const;
     void setUser(const QString &newUser);
 
     QString domain() const;
     void setDomain(const QString &newDomain);
 
     QString password() const;
     void setPassword(const QString &newPassword);
 
     RdpHostPreferences *preferences() const;
     void setHostPreferences(RdpHostPreferences *preferences);
 
     QSize size() const;
     void setSize(QSize size);
     Q_SIGNAL void sizeChanged();
 
     int port() const;
     void setPort(int port);
 
     bool start();
     void stop();
 
     bool sendEvent(QEvent *event, QWidget *source);
 
     const QImage *videoBuffer() const;
 
     Q_SIGNAL void rectangleUpdated(const QRect &rectangle);
 
     Q_SIGNAL void errorMessage(const QString &title, const QString &message);
 
 private:
     friend BOOL preConnect(freerdp*);
     friend BOOL postConnect(freerdp*);
     friend void postDisconnect(freerdp*);
     friend BOOL authenticate(freerdp*, char**, char**, char**);
     friend DWORD verifyCertificate(freerdp*, const char*, UINT16 port,const char*, const char*, const char*, const char*, DWORD);
     friend DWORD verifyChangedCertificate(freerdp*, const char*, UINT16, const char*, const char*, const char*, const char*, const char* , const char*, const char*, DWORD);
     friend BOOL endPaint(rdpContext *);
     friend BOOL resizeDisplay(rdpContext *);
 
     void setState(State newState);
 
     bool onPreConnect();
     bool onPostConnect();
     void onPostDisconnect();
 
     bool onAuthenticate(char** username, char** password, char** domain);
     CertificateResult onVerifyCertificate(const Certificate &certificate);
     CertificateResult onVerifyChangedCertificate(const Certificate &oldCertificate, const Certificate &newCertificate);
 
     bool onEndPaint();
     bool onResizeDisplay();
 
     void run();
 
     void emitErrorMessage();
 
     RdpView *m_view;
 
     freerdp *m_freerdp = nullptr;
     RdpContext *m_context = nullptr;
 
     State m_state = State::Initial;
 
     QString m_user;
     QString m_domain;
     QString m_password;
     QString m_host;
     int m_port = -1;
     QSize m_size;
 
     std::thread m_thread;
 
     QImage m_videoBuffer;
 
     RdpHostPreferences *m_preferences;
 };