;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;; ;; ;; ;; 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, "received: %s\n", receive_buffer mcall send, [socketnum], HandShake, 12, 0 DEBUGF 1, "Sending handshake: protocol version\n" call wait_for_data cmp dword[receive_buffer], 0x01000000 je no_security cmp dword[receive_buffer], 0x02000000 je vnc_security jmp err_security vnc_security: mov [status], STATUS_LOGIN call draw_gui no_security: mcall send, [socketnum], ClientInit, 1, 0 DEBUGF 1, "ClientInit sent\n" 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 mcall send, [socketnum], SetPixelFormat8, 20, 0 DEBUGF 1, "Sending pixel format\n" mcall send, [socketnum], SetEncodings, 12, 0 DEBUGF 1, "Sending encoding info\n" ; 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 1, "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 1, "\nunknown encoding: %u\n", eax jmp thread_loop next_rectangle: push esi call drawbuffer pop esi dec [rectangles] jnz rectangle_loop jmp request_fbu setcolourmapentries: DEBUGF 1, "Server sent SetColourMapEntries message\n" ; TODO 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 mcall recv, [socketnum], [datapointer], 4096, 0 pop edi esi edx ecx ebx cmp eax, -1 je err_disconnected add [datapointer], eax ; Check for buffer overflow cmp [datapointer], receive_buffer + RECEIVE_BUFFER_SIZE jbe @f ; Buffer is full, first needed data by program is pointed to by esi. ; Move all remaining data, starting from esi, 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 mov esi, receive_buffer @@: cmp eax, 4096 je .more ret wait_for_data: ; FIXME: add timeout mcall recv, [socketnum], receive_buffer, 4096, 0 cmp eax, -1 je err_disconnected test eax, eax jz wait_for_data 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