VncClientWidget.h

Go to the documentation of this file.

#pragma once
#include "VNCInterface/Headers/VncDefines.h"
#include "NDEVR/Headers/NDEVRMouseManager.h"
#include <NDEVR/Device.h>
#include "Base/Headers/BaseValues.hpp"
#include "Base/Headers/String.h"
#include "Base/Headers/Time.h"
#include <QWidget>
#include <QtWidgets>
#include <QtNetwork>
#include <QRegion>
#include <QKeyEvent>
namespace NDEVR
{
    class TcpServer;
    struct VncScreen
    {
        uint04 id;
        QRect rect;
        uint04 flags;
    };
    class VncClientWidget;
    class VncDevice : public Device
    {
        Q_OBJECT
    public:
        explicit VncDevice(VncClientWidget* parent);
        ~VncDevice();
        void update() override;
        [[nodiscard]] virtual TimeSpan updateInterval() const override;
        bool connectToVncServer(const String& ip, const String& password, int port = 5900);
        bool connectToVncServer();
        bool isConnectedToServer();
        void disconnectFromVncServer();
        void requestFullRefresh();
        
        
        bool postEvent(QEvent* event);
        void postKeyEvent(QKeyEvent* event, bool is_down);
        
    protected:
        void waitToRead(uint04 size) const;
        bool sendCommand(const char* const command, uint04 min_response_size, uint04 retry_count);
        
        void sendEncodings();
        void sendContinuousUpdateRequest(bool enable);
        void sendUpdateFrameBufferRequest(bool force_refresh_all);
        void sendRequestDesktopSize(QSize size, fltp08 scale_factor = 1.0);
        void sendMouseEvent(const MouseEvent& event);
        void sendKeyEvent(bool is_pressed, QKeyEvent* event);
        void sendKeyEvent(bool is_pressed, const KeyEvent& event);
        bool setupVncConnection(const String& password);
        void tearDownTcp();
        
        QByteArray desHash(QByteArray challenge, QString passStr);
        uint02 makeU16(quint8 l, quint8 h);
        uint04 makeU32(quint16 l, quint16 h);
        uint04 makeU32(quint8 lowest, quint8 low, quint8 high, quint8 highest);
        static quint32 TranslateRfbKey(int key, bool modifier);
        void readFrameBufferMessage();
        void setFrameBufferSize(const QSize& size);
    private slots:
        void onServerMessage();
    protected:
        NDEVRMouseManager m_mouse_manager;
        VncClientWidget* m_widget;
        QTcpSocket* m_socket;
        QSize m_frame_size;
        Time m_last_full_refresh_time;
        Time m_last_refresh_time;
        String m_password;
        String m_connection;
        Buffer<VncScreen> m_screens;
        PixelFormat m_pixel_format;
        Buffer<KeyEvent> m_key_events;
        Buffer<bool> m_key_down_events;
        BitFlag m_mouse_button_flag = BitFlag(0);
        QImage m_copy_image;
        int m_port;
 
        fltp04 m_jpg_quality;
        bool m_has_sent_frame_request = false;
        bool m_setup_connection = false;
        bool m_request_full_refresh = false;
        bool m_is_continuous_mode = false;
        bool m_supports_continuous_updates = false;
        bool m_supports_extended_size = false;
        bool m_request_disconnect = false;
        bool m_request_client_resize = true;
        bool m_is_ready = false;
        
    };
    class VncClientWidget : public QWidget
    {
        static constexpr uint01 s_screen_count = 2;
        Q_OBJECT
    public:
        explicit VncClientWidget(QWidget* parent = 0);
        ~VncClientWidget();
        void mousePressEvent(QMouseEvent* event) override;
        bool connectToVncServer(const String& ip, const String& password, int port = 5900);
        bool isConnectedToServer();
        void disconnectFromVncServer();
        void tryRefreshScreen(bool force_refresh_all);
        void setResizeTcpServer(TcpServer* server, fltp08 size_multiplier);
        void setMouse(const Vector<2, uint04>& hotspot, QImage mouse);
        QImage* imageToRender();
        void refreshTcpSize();
        void commit(const QRegion& region);
    protected:
        bool event(QEvent* event) override;
        void paintEvent(QPaintEvent*) override;
        void keyPressEvent(QKeyEvent* event) override;
        void keyReleaseEvent(QKeyEvent* event) override;
        void resizeEvent(QResizeEvent* event) override;
    private:
        VncDevice* m_device;
        TcpServer* m_tcp_server;
        fltp08 m_size_multiplier;
        QImage m_screen_buffer[s_screen_count];
        uint01 m_screen_index = 0;
        bool m_is_painting;
    };
}