Subversion Repositories Kolibri OS

server_parser:

; Commands are always 3 numbers and followed by a space

; If a server decides it needs multiline output,

; first lines will have a dash instead of space after numbers,

; thus they are simply ignored in this simple command parser.

        cmp     dword[buf_cmd+4], "Open"

        je      init_download_count

        cmp     dword[buf_cmd], "150 "

        je      data_loop

        cmp     dword[buf_cmd], "220 "

        je      connect_ok

;        cmp     dword[buf_cmd], "226 "

;        je      transfer_ok

        cmp     dword[buf_cmd], "227 "

        je      pasv_ok

        cmp     dword[buf_cmd], "230 "

        je      login_ok

;        cmp     dword[buf_cmd], "250"

;        je      op_ok

        cmp     dword[buf_cmd], "331 "

        je      pass

;        cmp     dword[buf_cmd], "421 "

;        je      timeout

        cmp     dword[buf_cmd], "503 "         ; login first

        je      welcome

        cmp     dword[buf_cmd], "530 "         ; password incorrect

        je      welcome

        cmp     dword[buf_cmd], "550 "

        je      close_datacon

        cmp     byte[buf_cmd+3], "-"

        je      wait_for_servercommand

        jmp     wait_for_usercommand

welcome:

        mov     [status], STATUS_CONNECTED

        jmp     wait_for_usercommand

connect_ok:

        mov     [status], STATUS_CONNECTED

        jmp     arg_handler.copy_user

pass:

        mov     [status], STATUS_NEEDPASSWORD

        jmp     wait_for_usercommand

login_ok:

        mov     [status], STATUS_LOGGED_IN

        cmp     [param_path], 0x20 ; no path specified

        jbe     wait_for_usercommand

        ; copy path to buf_cmd and execute CWD

        jmp     arg_handler.get_path

pasv_ok:

        sub     ecx, 4

        jb      .fail

        mov     al, "("

        mov     edi, buf_cmd + 4

        repne   scasb

        mcall   socket, AF_INET4, SOCK_STREAM, 0

        cmp     eax, -1

        jne     @f

        mov     eax, str_err_socket

        jmp     error

    @@: mov     [datasocket], eax

        mov     esi, edi

        call    ascii_dec

        mov     byte[sockaddr2.ip+0], bl

        call    ascii_dec

        mov     byte[sockaddr2.ip+1], bl

        call    ascii_dec

        mov     byte[sockaddr2.ip+2], bl

        call    ascii_dec

        mov     byte[sockaddr2.ip+3], bl

        call    ascii_dec

        mov     byte[sockaddr2.port+0], bl

        call    ascii_dec

        mov     byte[sockaddr2.port+1], bl

        icall   eax, interface_addr, interface.print, str_open

        mcall   connect, [datasocket], sockaddr2, 18

        cmp     eax, -1

        jne     @f

        mov     eax, str_err_connect

        jmp     error

    @@: jmp     wait_for_servercommand

  .fail:

        icall   eax, interface_addr, interface.print, str_unknown

        jmp     wait_for_servercommand

; get file size, initialize count for number of bytes downloaded

init_download_count:

        mov     edx, 0

        ; search for 'Open' in buf_cmd

        lea     esi, [buf_cmd+3]

      @@:

        inc     esi

        cmp     dword[esi], 'Open'

        je      @f

        cmp     byte[esi], 0

        jne     @b

        jmp     data_loop

      @@:

        ; get file size

        mov     al, '('

        mov     ecx, -1

        mov     edi, buf_cmd

        repne   scasb

        xor     eax, eax

        mov     ebx, 10

        xor     ecx, ecx

        mov     esi, edi

      @@:

        push    eax

        lodsb

        sub     al, '0'

        mov     cl, al

        pop     eax

        mul     ebx

        add     eax, ecx

        cmp     byte[esi], ' '

        jne     @b

        mov     [file_size], eax

data_loop:

        cmp     [operation], OPERATION_STOR

        je      .stor

        push    edx

; we are receiving data

        mcall   recv, [datasocket], buf_buffer2, BUFFERSIZE, 0

        pop     edx ; get byte count

        add     edx, eax

        test    ebx, ebx

        jnz     .done

        mov     byte[buf_buffer2 + eax], 0

        cmp     [operation], OPERATION_RETR

        je      .retr

        cmp     [operation], OPERATION_RDIR

        je      .rdir

        cmp     [operation], OPERATION_LIST

        je      .list

; not retreiving, just print to console

        icall   eax, interface_addr, interface.print, buf_buffer2

        jmp     data_loop

; for console, simply print. for gui, add name to tree list

  .list:

        ijmp    ebx, interface_addr, interface.list

; retreiving, save to file

  .retr:

        mov     [filestruct.ptr], buf_buffer2

        mov     [filestruct.size], eax

        push    eax

        mcall   70, filestruct

        test    eax, eax

        jz      @f

        call    error_fs

        jmp     close_datacon

      @@:

        pop     eax

        add     [filestruct.offset], eax

        icall   eax, interface_addr, interface.progress

        jmp     data_loop

; storing, send all data

  .stor:

        mcall   70, filestruct

        cmp     eax, 6          ; end of file

        je      .last_call

        test    eax, eax        ; error

        jz      @f

        call    error_fs

        jmp     close_datacon

      @@:

        add     [filestruct.offset], ebx

        mov     esi, ebx

        mcall   send, [datasocket], buf_buffer2, , 0

        mov     edx, [filestruct.offset]

        icall   eax, interface_addr, interface.progress

        jmp     .stor

  .last_call:

        mov     esi, ebx

        mcall   send, [datasocket], buf_buffer2, , 0

        mov     edx, [filestruct.offset]

        icall   eax, interface_addr, interface.progress

  .done:

        icall   eax, interface_addr, interface.print, str_close

        mcall   close, [datasocket]

        mov     [operation], OPERATION_NONE

        jmp     wait_for_servercommand

  .rdir:

        ; alloc/realloc memory block to store filenames

        mov     ecx, eax                        ; eax is size of buffer received

        inc     ecx

        add     ecx, [size_fname]               ; added old size to form new required size

        mcall   68, 20, , [ptr_fname]           ; realloc

        test    eax, eax

        je      error_heap

        mov     [ptr_fname], eax                ; eax contains the new block now

        mov     [ptr_queue], eax

        ; copy filenames into fname buffer

        mov     esi, buf_buffer2

        mov     edi, eax

        add     edi, [size_fname]

  .copy_buf:

        lodsb

        cmp     al, 13                          ; ignore any carriage return character

        je      .copy_buf

        stosb

        cmp     al, 10                          ; linefeed marks end of filename

        je      @f

        inc     [queued]

  @@:

        test    al, al                          ; 0 marks end of buffer

        jne     .copy_buf

        ; All received filenames have been copied, calculate new size of fname buffer

        dec     edi                             ; dont count the trailing 0 byte

        sub     edi, [ptr_fname]

        mov     [size_fname], edi

        jmp     data_loop

close_datacon:

        cmp     [operation], OPERATION_NONE

        je      wait_for_usercommand

        icall   eax, interface_addr, interface.print, str_close

        mcall   close, [datasocket]

        jmp     wait_for_usercommand

ascii_dec:

        xor     ebx, ebx

        mov     cl, 4                   ; max length is 3 digits + 1 separator

  .loop:

        lodsb

        sub     al, '0'

        jb      .done

        cmp     al, 9

        ja      .done

        lea     ebx, [ebx*4+ebx]        ; ebx *5

        shl     ebx, 1                  ; ebx *2

        add     bl, al

        dec     cl

        jnz     .loop

  .done:

        ret