Subversion Repositories Kolibri OS

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

;;                                                                 ;;

;; Copyright (C) KolibriOS team 2013-2014. All rights reserved.    ;;

;; Distributed under terms of the GNU General Public License       ;;

;;                                                                 ;;

;;  ftpc.asm - FTP client for KolibriOS                            ;;

;;                                                                 ;;

;;  Written by hidnplayr@kolibrios.org                             ;;

;;                                                                 ;;

;;          GNU GENERAL PUBLIC LICENSE                             ;;

;;             Version 2, June 1991                                ;;

;;                                                                 ;;

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

format binary as ""

TIMEOUT                 = 3     ; seconds

BUFFERSIZE              = 4096

STATUS_CONNECTING       = 0

STATUS_CONNECTED        = 1

STATUS_NEEDPASSWORD     = 2

STATUS_LOGGED_IN        = 3

OPERATION_NONE          = 0

OPERATION_LIST          = 1

OPERATION_RETR          = 2

OPERATION_STOR          = 3

OPERATION_RDIR          = 4

use32

; standard header

        db      'MENUET01'      ; signature

        dd      1               ; header version

        dd      start           ; entry point

        dd      i_end           ; initialized size

        dd      mem+0x1000      ; required memory

        dd      mem+0x1000      ; stack pointer

        dd      buf_cmd         ; parameters

        dd      path            ; path

include '../../macros.inc'

purge mov,add,sub

include '../../proc32.inc'

include '../../dll.inc'

include '../../network.inc'

include 'usercommands.inc'

include 'servercommands.inc'

start:

; initialize heap for using dynamic blocks

        mcall   68,11

        test    eax,eax

        je      exit2

; disable all events except network event

        mcall   40, EV_STACK

; load libraries

        stdcall dll.Load, @IMPORT

        test    eax, eax

        jnz     exit

; initialize console

        invoke  con_start, 1

        invoke  con_init, 80, 25, 80, 250, str_title

; find path to main settings file (ftpc.ini)

        mov     edi, path               ; Calculate the length of zero-terminated string

        xor     al, al

        mov     ecx, 1024

        repne   scasb

        dec     edi

        mov     esi, str_ini            ; append it with '.ini', 0

        movsd

        movsb

; get settings from ini

        invoke  ini.get_str, path, str_active, str_ip, str_active_ip, 16, 0

        mov     esi, str_active_ip

  .ip_loop:

        lodsb

        test    al, al

        jz      .ip_ok

        cmp     al, ' '

        je      .ip_ok

        cmp     al, '.'

        jne     .ip_loop

        mov     byte[esi-1], ','

        jmp     .ip_loop

  .ip_ok:

        mov     byte[esi-1], 0

        invoke  ini.get_int, path, str_active, str_port_start, 64000

        mov     [acti_port_start], ax

        invoke  ini.get_int, path, str_active, str_port_stop, 65000

        mov     [acti_port_stop], ax

        invoke  ini.get_str, path, str_general, str_dir, buf_buffer1, BUFFERSIZE, 0

        mcall   30, 1, buf_buffer1                      ; set working directory

; Check for parameters, if there are some, resolve the address right away

; TODO: parse ftp://user:password@server.com:port/folder/subfolder type urls.

        cmp     byte [buf_cmd], 0

        jne     resolve

main:

; Clear screen

        invoke  con_cls

; Welcome user

        invoke  con_write_asciiz, str_welcome

; write prompt (in green color)

        invoke  con_set_flags, 0x0a

        invoke  con_write_asciiz, str_prompt

; read string

        invoke  con_gets, buf_cmd, 256

; check for exit

        test    eax, eax

        jz      done

        cmp     byte [buf_cmd], 10

        jz      done

; reset color back to grey and print newline

        invoke  con_set_flags, 0x07

        invoke  con_write_asciiz, str_newline

resolve:

        mov     [sockaddr1.port], 21 shl 8

; delete terminating '\n'

        mov     esi, buf_cmd

  @@:

        lodsb

        cmp     al, ':'

        je      .do_port

        cmp     al, 0x20

        ja      @r

        mov     byte [esi-1], 0

        jmp     .done

  .do_port:

        xor     eax, eax

        xor     ebx, ebx

        mov     byte [esi-1], 0

  .portloop:

        lodsb

        cmp     al, 0x20

        jbe     .port_done

        sub     al, '0'

        jb      error_hostname

        cmp     al, 9

        ja      error_hostname

        lea     ebx, [ebx*4 + ebx]

        shl     ebx, 1

        add     ebx, eax

        jmp     .portloop

  .port_done:

        xchg    bl, bh

        mov     [sockaddr1.port], bx

  .done:

; Say to the user that we're resolving

        invoke  con_write_asciiz, str_resolve

        invoke  con_write_asciiz, buf_cmd

; resolve name

        push    esp     ; reserve stack place

        invoke  getaddrinfo, buf_cmd, 0, 0, esp

        pop     esi

; test for error

        test    eax, eax

        jnz     error_resolve

; write results

        invoke  con_write_asciiz, str8          ; ' (',0

        mov     eax, [esi+addrinfo.ai_addr]     ; convert IP address to decimal notation

        mov     eax, [eax+sockaddr_in.sin_addr] ;

        mov     [sockaddr1.ip], eax             ;

        invoke  inet_ntoa, eax                  ;

        invoke  con_write_asciiz, eax           ; print ip

        invoke  freeaddrinfo, esi               ; free allocated memory

        invoke  con_write_asciiz, str9          ; ')',10,0

; open the socket

        mcall   socket, AF_INET4, SOCK_STREAM, 0

        cmp     eax, -1

        je      error_socket

        mov     [controlsocket], eax

; connect to the server

        invoke  con_write_asciiz, str_connect

        mcall   connect, [controlsocket], sockaddr1, 18

        cmp     eax, -1

        je      error_connect

        mov     [status], STATUS_CONNECTING

; Tell the user we're waiting for the server now.

        invoke  con_write_asciiz, str_waiting

; Reset 'offset' variable, it's used by the data receiver

        mov     [offset], 0

wait_for_servercommand:

; Any commands still in our buffer?

        cmp     [offset], 0

        je      .receive                        ; nope, receive some more

        mov     esi, [offset]

        mov     edi, buf_cmd

        mov     ecx, [size]

        add     ecx, esi

        jmp     .byteloop

; receive socket data

  .receive:

        mcall   26, 9

        add     eax, TIMEOUT*100

        mov     [timeout], eax

  .receive_loop:

        mcall   23, 50          ; Wait for event with timeout

        mcall   26, 9

        cmp     eax, [timeout]

        jge     error_timeout

        mcall   recv, [controlsocket], buf_buffer1, BUFFERSIZE, MSG_DONTWAIT

        test    eax, eax

        jnz     .got_data

        cmp     ebx, EWOULDBLOCK

        jne     error_socket

        jmp     .receive_loop

  .got_data:

        mov     [offset], 0

; extract commands, copy them to "buf_cmd" buffer

        lea     ecx, [eax + buf_buffer1]         ; ecx = end pointer

        mov     esi, buf_buffer1                 ; esi = current pointer

        mov     edi, buf_cmd

  .byteloop:

        cmp     esi, ecx

        jae     wait_for_servercommand

        lodsb

        cmp     al, 10                          ; excellent, we might have a command

        je      .got_command

        cmp     al, 13                          ; just ignore this byte

        je      .byteloop

        stosb

        jmp     .byteloop

  .got_command:                                 ; we have a newline check if its a command

        cmp     esi, ecx

        je      .no_more_data

        mov     [offset], esi

        sub     ecx, esi

        mov     [size], ecx

        jmp     .go_cmd

  .no_more_data:

        mov     [offset], 0

  .go_cmd:

        lea     ecx, [edi - buf_cmd]                  ; length of command

        xor     al, al

        stosb

        invoke  con_set_flags, 0x03             ; change color

        invoke  con_write_asciiz, buf_cmd             ; print servercommand

        invoke  con_write_asciiz, str_newline

        invoke  con_set_flags, 0x07             ; reset color

        jmp     server_parser                   ; parse command

wait_for_usercommand:

; Are there any files in the transfer queue?

        cmp     [queued], 0

        ja      transfer_queued                 ; Yes, transfer those first.

; change color to green for user input

        invoke  con_set_flags, 0x0a

; If we are not yet connected, request username/password

        cmp     [status], STATUS_CONNECTED

        je      .connected

        cmp     [status], STATUS_NEEDPASSWORD

        je      .needpass

; write prompt

        invoke  con_write_asciiz, str_prompt

; read string

        invoke  con_gets, buf_cmd, 256

; print a newline and reset the color back to grey

        invoke  con_write_asciiz, str_newline

        invoke  con_set_flags, 0x07

        cmp     dword[buf_cmd], "cwd "

        je      cmd_cwd

        cmp     dword[buf_cmd], "mkd "

        je      cmd_mkd

        cmp     dword[buf_cmd], "rmd "

        je      cmd_rmd

        cmp     dword[buf_cmd], "pwd" + 10 shl 24

        je      cmd_pwd

        cmp     dword[buf_cmd], "bye" + 10 shl 24

        je      cmd_bye

        cmp     dword[buf_cmd], "rdir"

        je      cmd_rdir

        cmp     byte[buf_cmd+4], " "

        jne     @f

        cmp     dword[buf_cmd], "lcwd"

        je      cmd_lcwd

        cmp     dword[buf_cmd], "retr"

        je      cmd_retr

        cmp     dword[buf_cmd], "stor"

        je      cmd_stor

        cmp     dword[buf_cmd], "dele"

        je      cmd_dele

  @@:

        cmp     byte[buf_cmd+4], 10

        jne     @f

        cmp     dword[buf_cmd], "list"

        je      cmd_list

        cmp     dword[buf_cmd], "help"

        je      cmd_help

        cmp     dword[buf_cmd], "cdup"

        je      cmd_cdup

  @@:

; Uh oh.. unknown command, tell the user and wait for new input

        invoke  con_write_asciiz, str_unknown

        jmp     wait_for_usercommand

  .connected:

; request username

        invoke  con_write_asciiz, str_user

        mov     dword[buf_cmd], "USER"

        mov     byte[buf_cmd+4], " "

        jmp     .send

  .needpass:

; request password

        invoke  con_write_asciiz, str_pass

        mov     dword[buf_cmd], "PASS"

        mov     byte[buf_cmd+4], " "

        invoke  con_set_flags, 0x00             ; black text on black background for password

  .send:

; read string

        mov     esi, buf_cmd+5

        invoke  con_gets, esi, 256

; find end of string

        mov     edi, buf_cmd+5

        mov     ecx, 256

        xor     al, al

        repne   scasb

        lea     esi, [edi-buf_cmd]

        mov     word[edi-2], 0x0a0d

; and send it to the server

        mcall   send, [controlsocket], buf_cmd, , 0

        invoke  con_write_asciiz, str_newline

        invoke  con_set_flags, 0x07             ; reset color

        jmp     wait_for_servercommand

; files for rdir operation are queued

transfer_queued:

        mov     esi, [ptr_queue]                ; always pointing to current part of ptr_fname_start

        mov     edi, buf_cmd+5                  ; always point to filename for retr command

  .build_filename:

        lodsb

        stosb

        cmp     al, 10

        je      .get_file                       ; filename ends with character 10

        test    al, al

        jnz     .build_filename

        ; Error occured, we reached the end of the buffer before [queued] reached 0

        mov     [queued], 0

        mcall   68, 13, [ptr_fname]             ; free buffer

        test    eax, eax

        jz      error_heap

        jmp     wait_for_usercommand

  .get_file:

        mov     byte[edi], 0                    ; end filename with 0 byte

        mov     [ptr_queue], esi

        dec     [queued]

        jnz     cmd_retr

        mcall   68, 13, [ptr_fname]             ; free buffer

        test    eax, eax

        jz      error_heap

        jmp     cmd_retr

open_dataconnection:

        test    [mode], 1

        jnz     .active

        mcall   send, [controlsocket], str_PASV, str_PASV.length, 0

        ret

  .active:

        mcall   socket, AF_INET4, SOCK_STREAM, 0

        cmp     eax, -1

        je      error_socket

        mov     [datasocket], eax

        mov     ax, [acti_port_start]

        xchg    al, ah

        mov     [sockaddr2.port], ax

        mcall   bind, [datasocket], sockaddr2, 18

        cmp     eax, -1

        je      error_socket

        mcall   listen, [datasocket], 1

        cmp     eax, -1

        je      error_socket

        mov     dword[buf_buffer1], 'PORT'

        mov     byte[buf_buffer1+4], ' '

        mov     edi, buf_buffer1+5

        mov     esi, str_active_ip

  .loop:

        lodsb

        test    al, al

        jz      .ip_ok

        stosb

        jmp     .loop

  .ip_ok:

        mov     al, ','

        stosb

        movzx   eax, byte[sockaddr2.port+0]

        call    dword_ascii

        mov     al, ','

        stosb

        movzx   eax, byte[sockaddr2.port+1]

        call    dword_ascii

        mov     ax, 0x0a0d

        stosw

        lea     esi, [edi - buf_buffer1]

        mcall   send, [controlsocket], buf_buffer1, , 0

        mcall   accept, [datasocket], sockaddr2, 18        ; time to accept the awaiting connection..

        cmp     eax, -1

        je      error_socket

        push    eax

        mcall   close, [datasocket]

        pop     [datasocket]

        mcall   recv, [controlsocket], buf_buffer1, BUFFERSIZE, 0

        ret

; eax = input

; edi = ptr where to write

dword_ascii:

        push    edx ebx ecx

        mov     ebx, 10

        xor     ecx, ecx

       @@:

        xor     edx, edx

        div     ebx

        add     edx, '0'

        push    dx

        inc     ecx

        test    eax, eax

        jnz     @r

       @@:

        pop     ax

        stosb

        dec     ecx

        jnz     @r

        pop     ecx ebx edx

        ret

error_hostname:

        invoke  con_set_flags, 0x0c                     ; print errors in red

        invoke  con_write_asciiz, str_err_host

        jmp     wait_for_keypress

error_connect:

        invoke  con_set_flags, 0x0c                     ; print errors in red

        invoke  con_write_asciiz, str_err_connect

        jmp     wait_for_keypress

error_timeout:

        invoke  con_set_flags, 0x0c                     ; print errors in red

        invoke  con_write_asciiz, str_err_timeout

        jmp     wait_for_keypress

error_socket:

        invoke  con_set_flags, 0x0c                     ; print errors in red

        invoke  con_write_asciiz, str_err_socket

        jmp     wait_for_keypress

error_resolve:

        invoke  con_set_flags, 0x0c                     ; print errors in red

        invoke  con_write_asciiz, str_err_resolve

        jmp     wait_for_keypress

error_heap:

        invoke  con_set_flags, 0x0c                     ; print errors in red

        invoke  con_write_asciiz, str_err_heap

wait_for_keypress:

        invoke  con_set_flags, 0x07                     ; reset color to grey

        invoke  con_write_asciiz, str_push

        invoke  con_getch2

        mcall   close, [controlsocket]

        jmp     main

done:

        invoke  con_exit, 1

exit:

        mcall   close, [controlsocket]

exit2:

        mcall   -1

; data

str_title       db 'FTP client',0

str_welcome     db 'FTP client for KolibriOS v0.12',10

                db 10

                db 'Please enter ftp server address.',10,0

str_prompt      db '> ',0

str_resolve     db 'Resolving ',0

str_newline     db 10,0

str_err_resolve db 10,'Name resolution failed.',10,0

str_err_socket  db 10,'Socket error.',10,0

str_err_heap    db 10,'Cannot allocate memory from heap.',10,0

str_err_timeout db 10,'Timeout - no response from server.',10,0

str_err_connect db 10,'Cannot connect to the server.',10,0

str_err_host    db 10,'Invalid hostname.',10,0

str8            db ' (',0

str9            db ')',10,0

str_push        db 'Push any key to continue.',0

str_connect     db 'Connecting...',10,0

str_waiting     db 'Waiting for welcome message.',10,0

str_user        db "username: ",0

str_pass        db "password: ",0

str_unknown     db "Unknown command or insufficient parameters - type help for more information.",10,0

str_lcwd        db "Local working directory is now: ",0

str_open        db "opening data socket",10,0

str_close       db 10,"closing data socket",10,0

str_dot         db '.',0

str_help        db "available commands:",10

                db 10

                db "bye             - close the connection",10

                db "cdup            - change to parent of current directory on the server",10

                db "cwd  - change working directoy on the server",10

                db "dele      - delete file from the server",10

                db "list            - list files and folders in current server directory",10

                db "lcwd      - change local working directory",10

                db "mkd  - make directory on the server",10

                db "pwd             - print server working directory",10

                db "retr      - retreive file from the server",10

                db "rmd  - remove directory from the server",10

                db "stor      - store file on the server",10

                    db "rdir            - retreive all files from current server dir",10

                db 10,0

str_ini         db '.ini', 0

str_active      db 'active', 0

str_port_start  db 'port_start', 0

str_port_stop   db 'port_stop', 0

str_ip          db 'ip', 0

str_dir         db 'dir', 0

str_general     db 'general', 0

queued          dd 0

mode            db 0    ; passive = 0, active = 1

; FTP strings

str_PASV        db 'PASV',13,10

.length = $ - str_PASV

sockaddr1:

        dw AF_INET4

.port   dw ?

.ip     dd ?

        rb 10

sockaddr2:

        dw AF_INET4

.port   dw ?

.ip     dd ?

        rb 10

; import

align 4

@IMPORT:

library network, 'network.obj', console, 'console.obj', libini, 'libini.obj'

import  network,        \

        getaddrinfo,    'getaddrinfo',  \

        freeaddrinfo,   'freeaddrinfo', \

        inet_ntoa,      'inet_ntoa'

import  console,        \

        con_start,      'START',        \

        con_init,       'con_init',     \

        con_write_asciiz,'con_write_asciiz',     \

        con_exit,       'con_exit',     \

        con_gets,       'con_gets',\

        con_cls,        'con_cls',\

        con_getch2,     'con_getch2',\

        con_set_cursor_pos, 'con_set_cursor_pos',\

        con_write_string, 'con_write_string',\

        con_get_flags,  'con_get_flags', \

        con_set_flags,  'con_set_flags'

import  libini,         \

        ini.get_str,    'ini_get_str',\

        ini.get_int,    'ini_get_int'

i_end:

; uninitialised data

status          db ?

controlsocket   dd ?

datasocket      dd ?

offset          dd ?

size            dd ?

operation       dd ?

timeout         dd ?

ptr_fname       dd ?

size_fname      dd ?

ptr_queue       dd ?

acti_port_start dw ?

acti_port_stop  dw ?

acti_port       dw ?

str_active_ip   rb 16

filestruct:

  .subfn        dd ?

  .offset       dd ?

                dd ?

  .size         dd ?

  .ptr          dd ?

  .name         rb 1024

buf_buffer1     rb BUFFERSIZE+1

buf_buffer2     rb BUFFERSIZE+1

buf_cmd         rb 1024                 ; buffer for holding command string

path            rb 1024

mem: