vncrs 0.1.8

A pure Rust VNC server library for Windows
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70

use crate::error::{Result, VncError};
use crate::protocol::auth;
use crate::protocol::pixel_format::PixelFormat;
use log::info;
use std::io::{Read, Write};

pub fn perform_handshake<R: Read, W: Write>(
    reader: &mut R,
    writer: &mut W,
    width: u16,
    height: u16,
    name: &str,
    pixel_format: &PixelFormat,
    password: Option<&str>,
) -> Result<()> {
    writer.write_all(b"RFB 003.008\n")?;
    writer.flush()?;

    let mut version = [0u8; 12];
    reader.read_exact(&mut version)?;
    let version_str = String::from_utf8_lossy(&version);
    let trimmed = version_str.trim();
    info!("Client RFB version: {}", trimmed);

    if !trimmed.starts_with("RFB 003.") && !trimmed.starts_with("RFB 004.") {
        return Err(VncError::UnsupportedVersion(trimmed.to_string()));
    }

    if let Some(pass) = password {
        writer.write_all(&[1u8, 2u8])?;
        writer.flush()?;

        let mut chosen = [0u8; 1];
        reader.read_exact(&mut chosen)?;

        if chosen[0] != 2 {
            return Err(VncError::Handshake(format!(
                "Client chose unsupported security type: {}",
                chosen[0]
            )));
        }

        auth::perform_vnc_auth(reader, writer, pass)?;
    } else {
        writer.write_all(&[1u8, 1u8])?;
        writer.flush()?;

        let mut chosen = [0u8; 1];
        reader.read_exact(&mut chosen)?;

        writer.write_all(&0u32.to_be_bytes())?;
        writer.flush()?;
    }

    let mut shared_flag = [0u8; 1];
    reader.read_exact(&mut shared_flag)?;

    let name_bytes = name.as_bytes();
    let mut init = Vec::with_capacity(24 + name_bytes.len());
    init.extend_from_slice(&width.to_be_bytes());
    init.extend_from_slice(&height.to_be_bytes());
    init.extend_from_slice(&pixel_format.to_bytes());
    init.extend_from_slice(&(name_bytes.len() as u32).to_be_bytes());
    init.extend_from_slice(name_bytes);

    writer.write_all(&init)?;
    writer.flush()?;

    Ok(())
}