kolibrios-fun/programs/network/ftpc/servercommands.inc

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]

        ; refresh local directory list if in gui
        cmp     [operation], OPERATION_RETR
        jne     @f
        cmp     [interface_addr], gui
        jne     @f
        call    populate_local_tree_list
      @@:
        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