Subversion Repositories Kolibri OS

;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;

;;                                                                 ;;

;; 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                                ;;

;;                                                                 ;;

;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;

format binary as ""

__DEBUG__       = 1

__DEBUG_LEVEL__ = 2

BITS_PER_PIXEL  = 32            ; 8, 16, 24 or 32

SERVERADDRLEN   = 4096

use32

        org     0x0

        db      "MENUET01"      ; 8 byte id

        dd      0x01            ; header version

        dd      START           ; start of code

        dd      I_END           ; size of image

        dd      IM_END          ; memory for app

        dd      IM_END          ; esp

        dd      serveraddr, 0x0 ; I_Param , I_Path

include "../../macros.inc"

include "../../debug-fdo.inc"

include "../../proc32.inc"

include "../../dll.inc"

include "../../struct.inc"

include "../../develop/libraries/box_lib/trunk/box_lib.mac"

include "../../network.inc"

struct  PixelFormat

        bpp             db ?

        depth           db ?

        big_endian      db ?

        true_color      db ?

        red_max         dw ?

        green_max       dw ?

        blue_max        dw ?

        red_shift       db ?

        green_shift     db ?

        blue_shift      db ?

        padding         rb 3

ends

struct  ServerInit

        width           dw ?

        height          dw ?

        pixelformat     PixelFormat

        name_length     dd ?

        name            db ?

ends

xpos                    = 4

ypos                    = 21

TIMEOUT                 = 5             ; timeout in seconds

RECEIVE_BUFFER_SIZE     = 8*1024*1024   ; 8 Mib

STATUS_CONNECT          = 0

STATUS_CONNECTING       = 1

STATUS_REQ_LOGIN        = 2

STATUS_LOGIN            = 3

STATUS_CONNECTED        = 4

STATUS_CLOSED           = 5

STATUS_DISCONNECTED     = 10

STATUS_DNS_ERR          = 11

STATUS_SOCK_ERR         = 12

STATUS_CONNECT_ERR      = 13

STATUS_PROTO_ERR        = 14

STATUS_SECURITY_ERR     = 15

STATUS_LIB_ERR          = 16

STATUS_THREAD_ERR       = 17

STATUS_LOGIN_FAILED     = 18

STATUS_SECURITY_ERR_C   = 19

BYTES_PER_PIXEL = (BITS_PER_PIXEL + 7) / 8

include "keymap.inc"

include "gui.inc"

include "network.inc"

include "raw.inc"

include "copyrect.inc"

include "rre.inc"

include "trle.inc"

include "zrle.inc"

include "des.inc"

START:

        mcall   68, 11                  ; init heap

; Load libraries

        stdcall dll.Load, @IMPORT

        test    eax, eax

        jz      @f

        mov     [status], STATUS_LIB_ERR

  @@:

; Check if we got a server address through parameters

        cmp     byte[serveraddr], 0

        je      @f

        xor     al, al

        mov     edi, serveraddr

        mov     ecx, SERVERADDRLEN

        repne scasb

        sub     edi, serveraddr+1

        mov     [URLbox.size], edi

        mov     [URLbox.pos], edi

        call    open_connection

  @@:

; Present the user with the GUI and wait for network connection

        call    draw_gui

; Create main window

        mcall   71, 1, name             ; reset window caption (add server name)

        mov     edx, dword[screen]

        movzx   esi, dx

        shr     edx, 16

        add     edx, 2*xpos

        add     esi, ypos+xpos

        mcall   67, 10, 10              ; resize the window

        mcall   40, EVM_MOUSE + EVM_MOUSE_FILTER + EVM_KEY + EVM_REDRAW + EVM_BUTTON

        mcall   66, 1, 1                ; Switch keyboard to scancode mode

        call    generate_keymap

redraw:

        mcall   12, 1

        mov     ebx, dword[screen]

        movzx   ecx, bx

        shr     ebx, 16

        mov     edx, 0x74ffffff

        mov     edi, name

        mcall   0                       ; draw window

        mcall   12, 2

draw_framebuffer:

        DEBUGF  1, "Drawing framebuffer\n"

        mcall   7, framebuffer, dword[screen], 0

        mov     [update_framebuffer], 0

mainloop:

        cmp     [status], STATUS_CONNECTED

        jne     draw_gui

        cmp     [update_framebuffer], 0

        jne     draw_framebuffer

        mcall   23, 10                  ; Check for event with 0,1s timeout

        dec     eax

        jz      redraw

        dec     eax

        jz      key

        dec     eax

        jz      button

        sub     eax, 3

        jz      mouse

        jmp     mainloop

key:

        mcall   66, 3

        mov     ebx, eax        ; get modifier keys

        mcall   2               ; get key scancode

        cmp     ah, 224         ; extended keycode?

        je      .extended

        xor     al, al

        test    ah, 0x80        ; key up?

        jnz     @f

        inc     al

  @@:

        mov     byte[KeyEvent.down], al

        movzx   eax, ah

        test    ebx, 100000b    ; alt?

        jz      .no_alt

        mov     ax, [keymap_alt+eax*2]

        jmp     .key

  .no_alt:

        test    ebx, 11b        ; shift?

        jz      .no_shift

        mov     ax, [keymap_shift+eax*2]

        jmp     .key

  .no_shift:

        test    ebx, 10000000b  ; numlock ?

        jz      .no_numlock

        cmp     al, 71

        jb      .no_numlock

        cmp     al, 83

        ja      .no_numlock

        mov     ah, [keymap_numlock+eax-71]

        xor     al, al

        jmp     .key

  .extended:                    ; extended keys always use regular keymap

        mcall   2

        shr     eax, 8

        jz      mainloop

  .no_numlock:

        mov     ax, [keymap+eax*2]

  .key:

        test    ax, ax

        jz      mainloop

        mov     word[KeyEvent.key+2], ax

        DEBUGF  1, "Sending key: 0x%x\n", ax

        mcall   send, [socketnum], KeyEvent, 8, 0

        jmp     mainloop

mouse:

;        DEBUGF  1, "Sending pointer event\n"

        mcall   37, 1           ; get mouse pos

        bswap   eax

        mov     [PointerEvent.x], ax

        shr     eax, 16

        mov     [PointerEvent.y], ax

        mcall   37, 2           ; get mouse buttons

        test    al, 00000010b   ; test if right button was pressed  (bit 1 in kolibri)

        jz      @f

        add     al, 00000010b   ; in RFB protocol it is bit 2, so if we add bit 2 again, we"ll get bit 3 and bit 1 will remain the same

      @@:

        mov     [PointerEvent.mask], al

        mcall   send, [socketnum], PointerEvent, 6, 0

        jmp     mainloop

button:

        mcall   17              ; get id

        mov     [status], STATUS_CLOSED

        mcall   close, [socketnum]

        mcall   -1

; DATA AREA

include_debug_strings

keymap_numlock:

        db      '7', '8', '9', '-'

        db      '4', '5', '6', '+'

        db      '1', '2', '3'

        db      '0', '.'

HandShake               db "RFB 003.003", 10

ClientInit              db 0            ; not shared

SetPixelFormat32        db 0            ; setPixelformat

                        db 0, 0, 0      ; padding

.bpp                    db 32           ; bits per pixel

.depth                  db 24           ; depth

.big_endian             db 0            ; big-endian flag

.true_color             db 1            ; true-colour flag

.red_max                db 0, 255       ; red-max

.green_max              db 0, 255       ; green-max

.blue_max               db 0, 255       ; blue-max

.red_shift              db 16           ; red-shift

.green_shift            db 8            ; green-shift

.blue_shift             db 0            ; blue-shift

                        db 0, 0, 0      ; padding

SetPixelFormat24        db 0            ; setPixelformat

                        db 0, 0, 0      ; padding

.bpp                    db 24           ; bits per pixel

.depth                  db 24           ; depth

.big_endian             db 0            ; big-endian flag

.true_color             db 1            ; true-colour flag

.red_max                db 0, 255       ; red-max

.green_max              db 0, 255       ; green-max

.blue_max               db 0, 255       ; blue-max

.red_shift              db 16           ; red-shift

.green_shift            db 8            ; green-shift

.blue_shift             db 0            ; blue-shift

                        db 0, 0, 0      ; padding

SetPixelFormat16        db 0            ; setPixelformat

                        db 0, 0, 0      ; padding

.bpp                    db 16           ; bits per pixel

.depth                  db 16           ; depth

.big_endian             db 0            ; big-endian flag

.true_color             db 1            ; true-colour flag

.red_max                db 0, 31        ; red-max

.green_max              db 0, 63        ; green-max

.blue_max               db 0, 31        ; blue-max

.red_shift              db 11           ; red-shift

.green_shift            db 5            ; green-shift

.blue_shift             db 0            ; blue-shift

                        db 0, 0, 0      ; padding

SetPixelFormat8         db 0            ; setPixelformat

                        db 0, 0, 0      ; padding

.bpp                    db 8            ; bits per pixel

.depth                  db 8            ; depth

.big_endian             db 0            ; big-endian flag

.true_color             db 1            ; true-colour flag

.red_max                db 0, 7         ; red-max

.green_max              db 0, 7         ; green-max

.blue_max               db 0, 3         ; blue-max

.red_shift              db 0            ; red-shift

.green_shift            db 3            ; green-shift

.blue_shift             db 6            ; blue-shift

                        db 0, 0, 0      ; padding

SetEncodings            db 2            ; setEncodings

                        db 0            ; padding

                        db 0, 4         ; number of encodings

;                        db 0, 0, 0, 16  ; ZRLE

                        db 0, 0, 0, 15  ; TRLE

                        db 0, 0, 0, 2   ; RRE

                        db 0, 0, 0, 1   ; Copyrect encoding

                        db 0, 0, 0, 0   ; raw encoding

  .length = $ - SetEncodings

FramebufferUpdateRequest        db 3

.inc                            db 0    ; incremental

.x                              dw 0

.y                              dw 0

.width                          dw 0

.height                         dw 0

KeyEvent                db 4            ; keyevent

.down                   db 0            ; down-flag

                        dw 0            ; padding

.key                    dd 0            ; key

PointerEvent            db 5            ; pointerevent

.mask                   db 0            ; button-mask

.x                      dw 0            ; x-position

.y                      dw 0            ; y-position

sockaddr1:

                dw AF_INET4

.port           dw 0x0c17               ; 5900

.ip             dd 0

                rb 10

beep            db 0x85, 0x25, 0x85, 0x40, 0

status                  dd STATUS_CONNECT

update_gui              dd 0

mouse_dd                dd 0

update_framebuffer      dd 0

thread_id               dd 0

deflate_buffer          dd 0

deflate_length          dd ?

deflate_str             dd ?

URLbox          edit_box 235, 70, 20, 0xffffff, 0x6f9480, 0, 0, 0, SERVERADDRLEN, serveraddr, mouse_dd, ed_focus, 0, 0

USERbox         edit_box 215, 90, 10, 0xffffff, 0x6f9480, 0, 0, 0, 127, username, mouse_dd, ed_focus, 0, 0

PASSbox         edit_box 215, 90, 30, 0xffffff, 0x6f9480, 0, 0, 0, 127, password, mouse_dd, ed_pass, 0, 0

serverstr       db "server:"

userstr         db "username:"

passstr         db "password:"

connectstr      db "Connect"

loginstr        db "Log in"

cancelstr       db "Cancel"

okstr           db "OK"

okstr_e:

sz_err_disconnected     db "The server has closed the connection unexpectedly.", 0

sz_err_dns              db "Could not resolve hostname.", 0

sz_err_sock             db "Could not open socket.", 0

sz_err_connect          db "Could not connect to the server.", 0

sz_err_proto            db "A protocol error has occured.", 0

sz_err_security         db "An authentication problem has occured.", 0

sz_err_library          db "Could not load needed libraries.", 0

sz_err_thread           db "Could not create thread.", 0

sz_err_login_failed     db "Login failed.", 0

err_msg         dd sz_err_disconnected

                dd sz_err_dns

                dd sz_err_sock

                dd sz_err_connect

                dd sz_err_proto

                dd sz_err_security

                dd sz_err_library

                dd sz_err_thread

                dd sz_err_login_failed

                dd sz_err_security_c

; import

align 4

@IMPORT:

library network,                "network.obj",\

        box_lib,                "box_lib.obj",\

        archiver,               "archiver.obj"

import  network,\

        getaddrinfo,            "getaddrinfo",  \

        freeaddrinfo,           "freeaddrinfo", \

        inet_ntoa,              "inet_ntoa"

import  box_lib,\

        edit_box_draw,          "edit_box",\

        edit_box_key,           "edit_box_key",\

        edit_box_mouse,         "edit_box_mouse",\

        scrollbar_v_draw,       "scrollbar_v_draw",\

        scrollbar_v_mouse,      "scrollbar_v_mouse",\

        scrollbar_h_draw,       "scrollbar_h_draw",\

        scrollbar_h_mouse,      "scrollbar_h_mouse"

import  archiver,\

        deflate_unpack2,        "deflate_unpack2"

name                    db "VNC viewer "

.dash                   db 0, " "

I_END:

servername              rb 64+1

serveraddr              db 0

                        rb SERVERADDRLEN

socketnum               dd ?

datapointer             dd ?

rectangles              dw ?

rectangle:

.x                      dd ?

.y                      dd ?

.width                  dd ?

.height                 dd ?

subrectangles           dd ?

subrectangle:

.x                      dd ?

.y                      dd ?

.width                  dd ?

.height                 dd ?

screen:                 ; Remote screen resolution

.height                 dw ?

.width                  dw ?

palette                 rd 128  ; TRLE/ZRLE

palettesize             db ?    ; TRLE/ZRLE

keymap                  rw 128

keymap_shift            rw 128

keymap_alt              rw 128

username                rb 128

password                rb 128

keys                    rd 32*2         ; DES keys for VNC authentication

sz_err_security_c       rb 512+1

receive_buffer          rb RECEIVE_BUFFER_SIZE

framebuffer             rb 1280*1024*3  ; framebuffer

                        rb 0x1000

thread_stack:

                        rb 0x1000

IM_END: