forked from KolibriOS/kolibrios
450 lines
12 KiB
PHP
450 lines
12 KiB
PHP
;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
|
|
;; ;;
|
|
;; 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
|
|
|
|
; resolve name
|
|
push esp ; reserve stack place
|
|
invoke getaddrinfo, serveraddr, 0, 0, esp
|
|
pop esi
|
|
; test for error
|
|
test eax, eax
|
|
jnz err_dns
|
|
|
|
mov eax, [esi+addrinfo.ai_addr]
|
|
mov eax, [eax+sockaddr_in.sin_addr]
|
|
mov [sockaddr1.ip], eax
|
|
|
|
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
|
|
|
|
invoke freeaddrinfo, esi
|
|
|
|
mcall socket, AF_INET4, SOCK_STREAM, 0
|
|
cmp eax, -1
|
|
je err_sock
|
|
|
|
mov [socketnum], eax
|
|
mcall connect, [socketnum], sockaddr1, 18
|
|
cmp eax, -1
|
|
je err_connect
|
|
|
|
; TODO: implement timeout
|
|
call wait_for_data
|
|
|
|
cmp dword[receive_buffer], "RFB "
|
|
jne err_proto
|
|
DEBUGF 1, "Sending handshake\n"
|
|
mcall send, [socketnum], HandShake, 12, 0
|
|
call wait_for_data
|
|
|
|
cmp dword[receive_buffer], 0x01000000 ; no security
|
|
je initialize
|
|
cmp dword[receive_buffer], 0x02000000 ; VNC security
|
|
je vnc_security
|
|
|
|
jmp err_security
|
|
|
|
vnc_security:
|
|
|
|
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 ebx, dword[receive_buffer+4]
|
|
mov edx, dword[receive_buffer+8]
|
|
call encrypt_DES
|
|
mov dword[receive_buffer+4], ebx
|
|
mov dword[receive_buffer+8], edx
|
|
|
|
mov ebx, dword[receive_buffer+12]
|
|
mov edx, dword[receive_buffer+16]
|
|
call encrypt_DES
|
|
mov dword[receive_buffer+12], ebx
|
|
mov dword[receive_buffer+16], 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
|
|
|
|
mcall send, [socketnum], receive_buffer+4, 16, 0
|
|
|
|
call wait_for_data
|
|
cmp dword[receive_buffer], 0
|
|
jne err_login
|
|
; jmp initialize
|
|
|
|
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, 12, 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 update
|
|
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
|
|
DEBUGF 1, "rectangle: width=%u height=%u x=%u y=%u encoding: ",\
|
|
[rectangle.width]:2, [rectangle.height]:2, [rectangle.x]:2, [rectangle.y]:2
|
|
|
|
cmp eax, 0
|
|
je encoding_raw
|
|
cmp eax, 1
|
|
je encoding_CopyRect
|
|
cmp eax, 2
|
|
je encoding_RRE
|
|
; cmp eax, 5
|
|
; je encoding_hextile
|
|
; 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:
|
|
; 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
|
|
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_security:
|
|
mov [status], STATUS_SECURITY_ERR
|
|
inc [update_gui]
|
|
mcall -1
|
|
ret
|
|
|
|
err_login:
|
|
mov [status], STATUS_LOGIN_FAILED
|
|
inc [update_gui]
|
|
mcall -1
|
|
ret
|