;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
;; ;;
;; Copyright (C) KolibriOS team 2010-2015. All rights reserved. ;;
;; Distributed under terms of the GNU General Public License ;;
;; ;;
;; VNC client for KolibriOS ;;
;; ;;
;; Written by hidnplayr@kolibrios.org ;;
;; ;;
;; GNU GENERAL PUBLIC LICENSE ;;
;; Version 2, June 1991 ;;
;; ;;
;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
thread_start:
mcall 40, 0 ; disable all events for this thread
; Extract port number from server address
mov esi, serveraddr
@@:
lodsb
test al, al
jz .port_done
cmp al, ':'
jne @r
mov byte[esi-1], 0 ; replace colon with 0 byte, we dont want to upset getaddrinfo
xor eax, eax
xor ebx, ebx ; port number
@@:
lodsb
test al, al
jz @f
sub al, '0'
jb err_dns
cmp al, 9
ja err_dns
lea ebx, [ebx*4+ebx]
lea ebx, [ebx*2+eax]
jmp @b
@@:
xchg bl, bh
mov [sockaddr1.port], bx
.port_done:
; Resolve hostname
push esp ; reserve stack place
invoke getaddrinfo, serveraddr, 0, 0, esp
pop esi
test eax, eax
jnz err_dns
mov eax, [esi+addrinfo.ai_addr]
mov eax, [eax+sockaddr_in.sin_addr]
mov [sockaddr1.ip], eax
invoke freeaddrinfo, esi
DEBUGF 1, "Connecting to %u.%u.%u.%u:%u\n", \
[sockaddr1.ip]:1, [sockaddr1.ip+1]:1, [sockaddr1.ip+2]:1, [sockaddr1.ip+3]:1, \
[sockaddr1.port]:2
; Open socket
mcall socket, AF_INET4, SOCK_STREAM, 0
cmp eax, -1
je err_sock
mov [socketnum], eax
; Connect to the server
mcall connect, [socketnum], sockaddr1, 18
cmp eax, -1
je err_connect
; Verify handshake from server
call wait_for_data
cmp dword[receive_buffer], "RFB "
jne err_proto
add esi, 12
; Did we get an error message already?
cmp eax, 16
jb @f
lodsd
test eax, eax
je err_handshake
@@:
; Reply to handshake
DEBUGF 1, "Sending handshake\n"
mcall send, [socketnum], HandShake, 12, 0
; VNC 3.3 protocol: server decides security type
call wait_for_data
lodsd
cmp eax, 0x00000000
je err_handshake
cmp eax, 0x01000000 ; no security
je initialize
cmp eax, 0x02000000 ; VNC security
je vnc_security
jmp err_proto
vnc_security:
push esi ; pointer to message
mov dword[password], 0
mov dword[password+4], 0
and [USERbox.flags], not ed_focus
or [USERbox.flags], ed_disabled
or [PASSbox.flags], ed_focus
mov [status], STATUS_REQ_LOGIN
inc [update_gui]
@@:
mcall 5, 10
cmp [status], STATUS_LOGIN
je @f
cmp [status], STATUS_REQ_LOGIN
je @r
mcall -1
@@:
DEBUGF 1, "VNC authentication\n"
; Bit reverse the password and create DES keys
mov ebx, dword[password]
mov edx, ebx
and ebx, 0xf0f0f0f0
shr ebx, 4
and edx, 0x0f0f0f0f
shl edx, 4
or ebx, edx
mov edx, ebx
and ebx, 0xCCCCCCCC
shr ebx, 2
and edx, 0x33333333
shl edx, 2
or ebx, edx
mov edx, ebx
and ebx, 0xAAAAAAAA
shr ebx, 1
and edx, 0x55555555
shl edx, 1
or ebx, edx
bswap ebx
mov eax, dword[password+4]
mov edx, eax
and eax, 0xf0f0f0f0
shr eax, 4
and edx, 0x0f0f0f0f
shl edx, 4
or eax, edx
mov edx, eax
and eax, 0xCCCCCCCC
shr eax, 2
and edx, 0x33333333
shl edx, 2
or eax, edx
mov edx, eax
and eax, 0xAAAAAAAA
shr eax, 1
and edx, 0x55555555
shl edx, 1
or edx, eax
bswap edx
mov edi, keys
call DES_create_keys
; Encrypt message with DES
mov esi, [esp]
mov ebx, dword[esi+0]
mov edx, dword[esi+4]
call encrypt_DES
mov esi, [esp]
mov dword[esi+0], ebx
mov dword[esi+4], edx
mov ebx, dword[esi+8]
mov edx, dword[esi+12]
call encrypt_DES
mov esi, [esp]
mov dword[esi+8], ebx
mov dword[esi+12], edx
; Blank out the password and key fields in RAM
mov edi, password
mov ecx, 384/4
xor eax, eax
rep stosd
; Send the authentication response to server
pop edx
mcall send, [socketnum], , 16, 0
securityresult:
; Wait for SecurityResult from server
call wait_for_data
cmp dword[receive_buffer], 0 ; OK
jne err_login
initialize:
DEBUGF 1, "Sending ClientInit\n"
mcall send, [socketnum], ClientInit, 1, 0
call wait_for_data ; now the server should send init message
DEBUGF 1, "Serverinit: bpp: %u depth: %u bigendian: %u truecolor: %u\n", \
[receive_buffer+framebuffer.pixelformat.bpp]:1, \
[receive_buffer+framebuffer.pixelformat.depth]:1, \
[receive_buffer+framebuffer.pixelformat.big_endian]:1, \
[receive_buffer+framebuffer.pixelformat.true_color]:1
mov eax, dword[receive_buffer+framebuffer.width]
mov dword[FramebufferUpdateRequest.width], eax
bswap eax
mov dword[screen], eax
DEBUGF 1, "Screen width=%u, height=%u\n", [screen.width]:2, [screen.height]:2
DEBUGF 1, "Sending pixel format\n"
if BITS_PER_PIXEL = 8
mcall send, [socketnum], SetPixelFormat8, 20, 0
else if BITS_PER_PIXEL = 16
mcall send, [socketnum], SetPixelFormat16, 20, 0
else
mcall send, [socketnum], SetPixelFormat24, 20, 0
end if
DEBUGF 1, "Sending encoding info\n"
mcall send, [socketnum], SetEncodings, SetEncodings.length, 0
; Set main window caption to servername
mov ecx, dword[receive_buffer+framebuffer.name_length]
bswap ecx
cmp ecx, 64
jbe @f
mov ecx, 64
@@:
lea esi, [receive_buffer+framebuffer.name]
mov edi, servername
rep movsb
mov byte[edi], 0
mov [name.dash], "-"
; Tell the main thread we are ready for business!
mov [status], STATUS_CONNECTED
; Request initial framebuffer update from server
mov [FramebufferUpdateRequest.inc], 0
request_fbu:
DEBUGF 1, "Requesting framebuffer update\n"
mcall send, [socketnum], FramebufferUpdateRequest, 10, 0
mov [FramebufferUpdateRequest.inc], 1
thread_loop:
call read_data ; Read the data into the buffer
lodsb
cmp al, 0
je framebufferupdate
cmp al, 1
je setcolourmapentries
cmp al, 2
je bell
cmp al, 3
je servercuttext
DEBUGF 2, "Unknown server command: %u\n", al
jmp thread_loop
framebufferupdate:
@@:
lea eax, [esi+6]
cmp [datapointer], eax
jae @f
call read_data.more
jmp @b
@@:
inc esi ; padding
lodsw
xchg al, ah
mov [rectangles], ax
DEBUGF 1, "Framebufferupdate: %u rectangles\n", ax
rectangle_loop:
@@:
lea eax, [esi+12]
cmp [datapointer], eax
jae @f
call read_data.more
jmp @b
@@:
xor eax, eax
lodsw
xchg al, ah
mov [rectangle.x], eax
lodsw
xchg al, ah
mov [rectangle.y], eax
lodsw
xchg al, ah
mov [rectangle.width], eax
lodsw
xchg al, ah
mov [rectangle.height], eax
lodsd ; encoding
bswap eax
DEBUGF 1, "Rectangle: x=%u y=%u width=%u height=%u encoding: ",\
[rectangle.x]:2, [rectangle.y]:2, [rectangle.width]:2, [rectangle.height]:2
cmp eax, 0
je encoding_raw
cmp eax, 1
je encoding_CopyRect
cmp eax, 2
je encoding_RRE
cmp eax, 15
je encoding_TRLE
cmp eax, 16
je encoding_ZRLE
DEBUGF 2, "unknown encoding: %u\n", eax
jmp thread_loop
next_rectangle:
inc [update_framebuffer]
dec [rectangles]
jnz rectangle_loop
jmp request_fbu
setcolourmapentries:
DEBUGF 1, "Server sent SetColourMapEntries message\n"
@@:
lea eax, [esi+5]
cmp [datapointer], eax
jae @f
call read_data.more
jmp @b
@@:
inc esi ; padding
xor eax, eax
lodsw ; first color (just ignore for now)
lodsw ; number of colors (use to find end of message)
xchg al, ah
lea eax, [eax*2+eax]
shl eax, 1
@@:
push eax
add eax, esi
cmp [datapointer], eax
jae @f
call read_data.more
pop eax
jmp @b
@@:
pop eax
add esi, eax ; Just skip it for now.
jmp thread_loop
bell:
mcall 55, 55, , , beep
jmp thread_loop
servercuttext:
DEBUGF 1, "Server cut text\n"
@@:
lea eax, [esi+7]
cmp [datapointer], eax
jae @f
call read_data.more
jmp @b
@@:
add esi, 3
lodsd
bswap eax
mov ecx, eax
@@:
lea eax, [esi+ecx]
cmp [datapointer], eax
jae @f
call read_data.more
jmp @b
@@:
; TODO: paste text to clipboard
DEBUGF 1, "%u bytes of text\n", ecx
add esi, ecx
jmp thread_loop
read_data:
mov [datapointer], receive_buffer
mov esi, receive_buffer
.more:
push ebx ecx edx esi edi
neg esi
add esi, receive_buffer + RECEIVE_BUFFER_SIZE
jz .buffer_end_reached
xor edi, edi
mcall recv, [socketnum], [datapointer]
pop edi esi edx ecx ebx
cmp eax, -1
je err_sock
test eax, eax
jz err_disconnected
add [datapointer], eax
ret
.buffer_end_reached:
DEBUGF 1, "end of buffer reached, re-organizing\n"
pop edi esi edx ecx ebx
; Buffer is full, first needed data by program is pointed to by esi.
; Move all usefull data to begin of buffer
cmp esi, receive_buffer
je err_proto
mov ecx, [datapointer]
sub ecx, esi
mov edi, receive_buffer
rep movsb
mov [datapointer], edi ; new end of data
mov esi, receive_buffer ; new start of data
jmp .more
wait_for_data: ; FIXME: add timeout
mcall recv, [socketnum], receive_buffer, 4096, 0 ; MSG_DONTWAIT
cmp eax, -1
je err_sock
test eax, eax
jz err_disconnected
mov esi, receive_buffer
ret
err_disconnected:
mov [status], STATUS_DISCONNECTED
inc [update_gui]
mcall -1
err_dns:
mov [status], STATUS_DNS_ERR
inc [update_gui]
mcall -1
err_sock:
mov [status], STATUS_SOCK_ERR
inc [update_gui]
mcall -1
err_connect:
mov [status], STATUS_CONNECT_ERR
inc [update_gui]
mcall -1
ret
err_proto:
mov [status], STATUS_PROTO_ERR
inc [update_gui]
mcall -1
ret
err_handshake:
mov [status], STATUS_SECURITY_ERR
lodsd ; Custom message from server?
test eax, eax
jz .no_msg
bswap eax
mov ecx, eax
cmp ecx, 512
jb @f
mov ecx, 512
@@:
mov edi, sz_err_security_c
rep movsb
mov byte[edi], 0
mov [status], STATUS_SECURITY_ERR_C
.no_msg:
inc [update_gui]
mcall -1
ret
err_login:
mov [status], STATUS_LOGIN_FAILED
inc [update_gui]
mcall -1
ret