Subversion Repositories Kolibri OS

        stack           rb 0

        home_dir        rb 1024         ; home directory in wich the user is locked, asciiz

        fpath           rb 1024*3       ; file path, combination of home_dir, work_dir and filename

                                        ; Will also be used to temporarily store username

        type            db ?    ; ASCII/EBDIC/IMAGE/..

        socketnum       dd ?    ; Commands socket

        state           dd ?    ; disconnected/logging in/logged in/..

        passivesocknum  dd ?    ; when in passive mode, this is the listening socket

        datasocketnum   dd ?    ; socket used for data transfers

        permissions     dd ?    ; read/write/execute/....

        buffer_ptr      dd ?

        pid             dd ?    ; Process id of the current thread

        datasock        sockaddr_in

        buffer          rb BUFFERSIZE

;------------------------------------------------

;

; Internal function wich uses the 'commands'

;  table to call an appropriate cmd_xx function.

;

; input: esi = ptr to ascii commands

;        ecx = number of bytes input

;        ebp = pointer to thread_data structure

;

; output: none

;

;------------------------------------------------

align 4

parse_cmd:                              ; esi must point to command

        cmp     byte [esi], 0x20        ; skip all leading characters

        inc     esi

        dec     ecx

        cmp     ecx, 3

        jb      .error

        jmp     parse_cmd

  .ok:

        cmp     byte [esi+3], 0x20

        ja      @f

        mov     byte [esi+3], 0

       @@:

        mov     eax, [esi]

        mov     edi, commands           ; list of commands to scan

  .scanloop:

        cmp     eax, [edi]

        je      .got_it

        add     edi, 5*4

        jne     .scanloop

  .error:

        jb      login_first

        sendFTP "500 Unsupported command"

        ret

  .got_it:

        jmp     dword [edi + 4 + eax]

commands:               ; all commands must be in uppercase

        dd 'ABOR', login_first, login_first, login_first, cmdABOR

;        dd 'APPE', login_first, login_first, login_first, cmd_APPE

        dd 'CDUP', login_first, login_first, login_first, cmdCDUP

        dd 'CWD',  login_first, login_first, login_first, cmdCWD

        dd 'DELE', login_first, login_first, login_first, cmdDELE

;        dd 'HELP', login_first, login_first, login_first, cmd_HELP

        dd 'LIST', login_first, login_first, login_first, cmdLIST

;        dd 'MDTM', login_first, login_first, login_first, cmd_MDTM

;        dd 'MKD',  login_first, login_first, login_first, cmd_MKD

;        dd 'MODE', login_first, login_first, login_first, cmd_MODE

;        dd 'NLST', login_first, login_first, login_first, cmdNLST

        dd 'NOOP', login_first, login_first, login_first, cmdNOOP

        dd 'PASS', cmdPASS.0,   cmdPASS    , cmdPASS.2,   cmdPASS.3

        dd 'PASV', login_first, login_first, login_first, cmdPASV

        dd 'PORT', login_first, login_first, login_first, cmdPORT

        dd 'PWD',  login_first, login_first, login_first, cmdPWD

        dd 'QUIT', cmdQUIT,     cmdQUIT,     cmdQUIT,     cmdQUIT

;        dd 'REIN', login_first, login_first, login_first, cmd_REIN

;        dd 'REST', login_first, login_first, login_first, cmd_REST

        dd 'RETR', login_first, login_first, login_first, cmdRETR

;        dd 'RMD', login_first, login_first, login_first, cmd_RMD

;        dd 'RNFR', login_first, login_first, login_first, cmd_RNFR

;        dd 'RNTO', login_first, login_first, login_first, cmd_RNTO

;        dd 'SITE', login_first, login_first, login_first, cmd_SITE

;        dd 'SIZE', login_first, login_first, login_first, cmd_SIZE

;        dd 'STAT', login_first, login_first, login_first, cmd_STAT

        dd 'STOR', login_first, login_first, login_first, cmdSTOR

;        dd 'STOU', login_first, login_first, login_first, cmd_STOU

;        dd 'STRU', login_first, login_first, login_first, cmd_STRU

        dd 'SYST', login_first, login_first, login_first, cmdSYST

        dd 'TYPE', login_first, login_first, login_first, cmdTYPE

        dd 'USER', cmdUSER,     cmdUSER,     cmdUSER,     cmdUSER.2

        db 0    ; end marker

endg

align 4

        sendFTP "530 Please login with USER and PASS"

        ret

align 4

        sendFTP "550 Permission denied"

        ret

align 4

        invoke  con_set_flags, 0x0c

        invoke  con_write_asciiz, str_sockerr

        invoke  con_set_flags, 0x07

        sendFTP "425 Can't open data connection"

align 4

        and     [ebp + thread_data.permissions], not ABORT

        mov     [ebp + thread_data.mode], MODE_NOTREADY

        invoke  file.close, ebx

        mcall   close, [ebp + thread_data.datasocketnum]

        sendFTP "530 Transfer aborted"

align 4

        call    ascii_to_byte

        cmp     byte [esi], cl

        jne     .err

        inc     esi

        call    ascii_to_byte

        cmp     byte [esi], cl

        jne     .err

        inc     esi

        shl     ebx, 16

        call    ascii_to_byte

        cmp     byte [esi], cl

        jne     .err

        inc     esi

        call    ascii_to_byte

        ror     ebx, 16

  .err:

        ret

align 4         ; esi = ptr to str, output in eax

        xor     eax, eax

  .loop:

        sub     bl, '0'

        jb      .done

        cmp     bl, 9

        ja      .done

        lea     eax, [eax*4 + eax]      ;

        shl     eax, 1                  ; eax = eax * 10

        add     eax, ebx

        inc     esi

        jmp     .loop

  .done:

        ret

align 4

        push    edx ebx ecx

        xor     ecx, ecx

       @@:

        div     ebx

        add     edx, '0'

        pushw   dx

        inc     ecx

        test    eax, eax

        jnz     @r

       @@:

        stosb

        dec     ecx

        jnz     @r

        pop     ecx ebx edx

align 4

        lea     edi, [ebp + thread_data.fpath]

        mov     ecx, 1024

  .loop1:

        lodsb

        cmp     al, 0x20

        jb      .next

        stosb

        loop    .loop1

  .next:

        cmp     byte[edi-1], '/'

        dec     edi

       @@:

        lea     esi, [ebp + thread_data.work_dir]

  .loop2:

        lodsb

        cmp     al, 0x20

        jb      .done

        stosb

        loop    .loop2

  .done:

        stosb

        ret

        inc     [pasv_port]

        mov     ax, [pasv_port]

        jb      .restart

        cmp     ax, [pasv_end]

        ja      .restart

        ret

  .restart:

        popw    [pasv_port]

        ret

        cmp     [ebp + thread_data.mode], MODE_PASSIVE_OK

; If we are in active mode, it's time to open the data socket..

        jne     .not_active

        mov     ecx, [ebp + thread_data.datasocketnum]

        lea     edx, [ebp + thread_data.datasock]

        mov     esi, sizeof.thread_data.datasock

        mcall   connect

        cmp     eax, -1

        je      .socketerror

        invoke  con_write_asciiz, str_datasock2

        ret

  .already_open:

        ret

  .socketerror:

        jmp     socketerror

; If we are still in passive_wait, it's time we accept an incomming call..

        cmp     [ebp + thread_data.mode], MODE_PASSIVE_WAIT

        jne     .socketerror

  .try_now:

        lea     edx, [ebp + thread_data.datasock]

        mov     esi, sizeof.thread_data.datasock

        mcall   accept

        cmp     eax, -1

        jne     .pasv_ok

        mov     [ebp + thread_data.mode], MODE_PASSIVE_FAILED                   ; assume that we will fail

        mcall   23, 200

        mcall   accept

        cmp     eax, -1

        je      .socketerror

  .pasv_ok:

        mov     [ebp + thread_data.datasocketnum], eax

        mov     [ebp + thread_data.mode], MODE_PASSIVE_OK

        mcall   close   ; [ebp + thread_data.passivesocknum]

        mov     [ebp + thread_data.passivesocknum], -1

        invoke  con_write_asciiz, str_datasock

        ret

;

; This command aborts the current filetransfer.

;

;------------------------------------------------

align 4

cmdABOR:

        or      [ebp + thread_data.permissions], ABORT

        ret

;------------------------------------------------

;

; Change the directory to move up one level.

;

;------------------------------------------------

align 4

cmdCDUP:

        test    [ebp + thread_data.permissions], PERMISSION_CD

        cmp     byte [ebp + thread_data.work_dir+1], 0  ; are we in "/" ?

; find the end of asciiz string work_dir

        xor     al, al

        lea     edi, [ebp + thread_data.work_dir]

        repne   scasb

; return 2 characters (right before last /)

        sub     edi, 3

; and now search backwards, for a '/'

        mov     al,'/'

        neg     ecx

        add     ecx, 1024

        std

        repne   scasb

        cld

; terminate the string here

        mov     byte[edi+2], 0

  .done:

        lea     eax, [ebp + thread_data.work_dir]

        invoke  con_write_asciiz, eax

        invoke  con_write_asciiz, str_newline

        sendFTP "250 Command succesul"

;------------------------------------------------

;

; Change Working Directory.

;

;------------------------------------------------

align 4

cmdCWD:

        test    [ebp + thread_data.permissions], PERMISSION_CD

; do we have enough parameters?

        jbe     .err

; get ready to copy the path

        mov     ecx, 1024

        lea     edi, [ebp + thread_data.work_dir]

; if received dir starts with '/', we will simply copy it

        cmp     byte [esi], '/'

        je      .copyloop

; Find the end of work_dir string.

  .find_zero:

        repne   scasb

        dec     edi

; and now append work_dir with received string

; scan for end byte, or '.'

        lodsb

        cmp     al, 0x20

        jb      .done

;;;        cmp     al, '.'         ; '..' means we must go up one dir   TODO

;;;        je      .up

        stosb

        loop    .copyloop

; now, now make sure it ends with '/', 0

        cmp     byte [edi-1], '/'

        je      @f

        mov     byte [edi], '/'

        inc     edi

       @@:

        mov     byte [edi], 0

; TODO: Check directory on disk

        invoke  con_write_asciiz, eax

        invoke  con_write_asciiz, str_newline

        sendFTP "250 Command succesful"

  .err:

        ret

;------------------------------------------------

;

; Delete a file from the server.

;

;------------------------------------------------

align 4

cmdDELE:

        test    [ebp + thread_data.permissions], PERMISSION_DELETE

        ; Create path

        jae     .err

        sub     ecx, 5

        call    create_path

        lea     esi, [ebp + thread_data.buffer + 5]

        mov     ecx, 1024

        cmp     byte [esi], '/'

        jne     .loop

        inc     esi

  .loop:

        lodsb

        cmp     al, 0x20

        jl      .done

        stosb

        loop    .loop

  .done:

        xor     al, al

        stosb

        lea     ebx, [ebp + thread_data.fpath]

        invoke  con_write_asciiz, str_newline

        ; called fs function

        push    ebx

        dec     esp

        mov     byte[esp], 0

        push    dword 0

        push    dword 0

        push    dword 0

        push    dword 0

        push    dword 8

        mov     ebx, esp

        mcall   70

        add     esp, 6*4 + 1

        test    eax, eax

        sendFTP "250 Command succesful"

.err:

        sendFTP "550 No such file"

        ret

;------------------------------------------------

;

; List the files in the current working directory.

;

;------------------------------------------------

align 4

cmdLIST:

        test    [ebp + thread_data.permissions], PERMISSION_EXEC

        call    open_datasock

; Create fpath from home_dir and work_dir

        lea     ebx, [ebp + thread_data.fpath]

        invoke  con_write_asciiz, str_newline

; Start the search

        test    eax, eax

        jz      .nosuchdir

        lea     edi, [ebp + thread_data.buffer]

        test    eax, eax        ; did we find a file?

        jz      .done

        mov     ebx, eax        ; yes, save the descripter in ebx

; first, convert the attributes

        jnz     .folder

        test    [ebx + FileInfoA.Attributes], FA_READONLY

        mov     eax, '-rw-'

        jmp     .attr

  .folder:

        stosd

        jmp     .attr

  .readonly:

        stosd

  .attr:

        stosd

        mov     ax, 'w-'

        stosw

        mov     al, ' '

        stosb

; now..

        stosw

; now write owner, everything is owned by FTP, woohoo!

        stosd

        stosd

; now the filesize in ascii

        call    dword_to_ascii

        mov     al, ' '

        stosb

; then date (month/day/year)

        cmp     eax, 12

        ja      @f

        mov     eax, [months - 4 + 4*eax]

        stosd

       @@:

        movzx   eax, [ebx + FileInfoA.DateModify.day]

        mov     al, ' '

        stosb

        movzx   eax, [ebx + FileInfoA.DateModify.year]

        mov     al, ' '

        stosb

; and last but not least, filename

        mov     ecx, 264

  .nameloop:

        lodsb

        test    al, al

        jz      .namedone

        stosb

        loop    .nameloop

; insert a cr lf

        mov     ax, 0x0a0d

        stosw

        test    [ebp + thread_data.permissions], ABORT          ; Did we receive ABOR command from client?

; check next file

        jmp     .parse_file

; close file desc

        invoke  file.find.close, ebx                            ; ebx is descriptor of last file, eax will be -1 !!

; append the string with a 0

        stosb

; Warn the client we're about to send the data

        sendFTP "150 Here it comes.."

        pop     esi

; and send it to the client

        lea     edx, [ebp + thread_data.buffer]                 ; buffer ptr

        sub     esi, edx                                        ; length

        xor     edi, edi                                        ; flags

        mcall   send

; close the data socket..

        mcall   close, [ebp + thread_data.datasocketnum]

        sendFTP "226 List OK"

  .nosuchdir:

        ret

;------------------------------------------------

;

; List the filenames of the files in the current working directory.

;

;------------------------------------------------

align 4

cmdNLST:

        test    [ebp + thread_data.permissions], PERMISSION_EXEC

        ; TODO: same as list but simpler output format

        ret

;------------------------------------------------

;

; No operation, just keep the connection alive.

;

;------------------------------------------------

align 4

cmdNOOP:

        sendFTP "200 Command OK"

;------------------------------------------------

;

; Second phase of login process, client provides password.

;

;------------------------------------------------

align 4

cmdPASS:

; read the password from users.ini

        lea     ebx, [ebp + thread_data.fpath]                  ; temp username

        invoke  ini.get_str, path2, ebx, str_pass, edi, 512, str_infinity

        test    eax, eax                                        ; unable to read password? fail!

        jnz     .incorrect

        cmp     dword [edi], -1                                 ; no key, section or file found.. fail!

        je      .incorrect

        cmp     byte [edi], 0                                   ; zero password? ok!

        je      .ok

        add     esi, 5

        jbe     .incorrect                                      ; no password given? but hey, we need one! fail..

; compare with received password

        cmp     byte [esi-1], 0x20                              ; printeable characters left?

        jae     .incorrect

        cmp     byte [edi-1], 0

        jne     .incorrect

  .ok:

        mov     [ebp + thread_data.permissions], eax

        invoke  con_write_asciiz, str_pass_ok

        sendFTP "230 You are now logged in"

        ret

  .2:

        invoke  con_write_asciiz, str_pass_err

        mov     [ebp + thread_data.state], STATE_CONNECTED      ; reset state

        sendFTP "530 Login incorrect"

        ret

  .0:

        ret

  .3:

        ret

;------------------------------------------------

;

; Initiate a passive dataconnection.

;

;------------------------------------------------

align 4

cmdPASV:

;        cmp     [ebp + thread_data.passivesocknum], -1

;        mcall   close, [ebp + thread_data.passivesocknum]       ; if there is still a socket open, close it

;       @@:

; Open a new TCP socket

        cmp     eax, -1

        je      socketerror

        mov     [ebp + thread_data.passivesocknum], eax

; Bind it to a known local port

        mov     [ebp + thread_data.datasock.sin_addr], 0

        mov     ecx, eax                                        ; passivesocketnum

        mov     esi, sizeof.thread_data.datasock

  .next_port:                                                   ; TODO: break the endless loop

        mov     ax, [pasv_port]

        xchg    al, ah

        mov     [ebp + thread_data.datasock.sin_port], ax

        mcall   bind

        je      .next_port

; And set it to listen!

        cmp     eax, -1

        je      socketerror

; Tell our thread we are ready to accept incoming calls

; Now tell the client where to connect to in this format:

; where a1.a2.a3.a4 is the IP address and p1*256+p2 is the port number.

; '227 Entering passive mode ('

        mov     ecx, str_227.length

        mov     esi, str_227

        rep movsb

; ip

        movzx   eax, byte [serverip]

        call    dword_to_ascii

        mov     al, ','

        stosb

        movzx   eax, byte [serverip+1]

        call    dword_to_ascii

        mov     al, ','

        stosb

        movzx   eax, byte [serverip+2]

        call    dword_to_ascii

        mov     al, ','

        stosb

        movzx   eax, byte [serverip+3]

        call    dword_to_ascii

        mov     al, ','

        stosb

; port

        movzx   eax, byte [ebp + thread_data.datasock.sin_port]

        call    dword_to_ascii

        mov     al, ','

        stosb

        movzx   eax, byte [ebp + thread_data.datasock.sin_port+1]

        call    dword_to_ascii

; ')', 13, 10, 0

        mov     eax, ')' + (0x000a0d shl 8)

        stosd

        lea     esi, [edi - thread_data.buffer - 1]      ; calculate length, do not cound the trailing 0 byte

        mov     ecx, [ebp + thread_data.socketnum]

        lea     edx, [ebp + thread_data.buffer]

        xor     edi, edi

        mcall   send

        invoke  con_write_asciiz, edx

        ret

        .length = $ - str_227

endg

;------------------------------------------------

;

; Print the current working directory.

;

;------------------------------------------------

align 4

cmdPWD:

        mov     dword [ebp + thread_data.buffer], '257 '

        lea     edi, [ebp + thread_data.buffer+5]

        mov     ecx, 1024

  .loop:

        lodsb

        or      al, al

        jz      .ok

        stosb

        dec     ecx

        jnz     .loop

  .ok:

        lea     esi, [edi - thread_data.buffer + 3]

        sub     esi, ebp

        mov     ecx, [ebp + thread_data.socketnum]

        lea     edx, [ebp + thread_data.buffer]

        xor     edi, edi

        mcall   send

; Print the new working dir on the console

        invoke  con_write_asciiz, eax

        invoke  con_write_asciiz, str_newline

        ret

;------------------------------------------------

;

; Initiate an active dataconnection.

;

;------------------------------------------------

align 4

cmdPORT:

; PORT a1,a2,a3,a4,p1,p2

; Convert the IP

        mov     cl, ','

        call    ip_to_dword

; And put it in datasock

        mov     [ebp + thread_data.datasock.sin_addr], ebx

; Now the same with portnumber

        call    ascii_to_byte

        mov     byte[ebp + thread_data.datasock.sin_port], al

        inc     esi

        call    ascii_to_byte

        mov     byte[ebp + thread_data.datasock.sin_port+1], al

; We will open the socket, but do not connect yet!

        mcall   socket, AF_INET4, SOCK_STREAM, 0

        cmp     eax, -1

        je      socketerror

        mov     [ebp + thread_data.datasocketnum], eax

        sendFTP "225 Data connection open"

;------------------------------------------------

;

; Close the connection with client.

;

;------------------------------------------------

align 4

cmdQUIT:

        sendFTP "221 Bye!"

        mcall   close, [ebp + thread_data.socketnum]

        add     esp, 4          ; get rid of call return address

;

; Retrieve a file from the ftp server.

;

;------------------------------------------------

align 4

cmdRETR:

        test    [ebp + thread_data.permissions], PERMISSION_READ

        cmp     ecx, 1024 + 5

        sub     ecx, 5

        call    open_datasock

        call    create_path

        lea     esi, [ebp + thread_data.buffer + 5]

        mov     ecx, 1024

        cmp     byte [esi], '/'

        jne     .loop

        inc     esi

  .loop:

        lodsb

        cmp     al, 0x20

        jl      .done

        stosb

        loop    .loop

  .done:

        xor     al, al

        stosb

        lea     ebx, [ebp + thread_data.fpath]

        invoke  con_write_asciiz, str_newline

        invoke  file.open, ebx, O_READ

        jz      .cannot_open

        push    eax

        pop     ebx

  .read_more:

        jnz     abort_transfer

        lea     eax, [ebp + thread_data.buffer]                 ; FIXME: use another buffer!! if we receive something on control connection now, we screw up!

        cmp     eax, -1

        je      .cannot_open                                    ; FIXME: this is not the correct error

        invoke  con_write_asciiz, str2

        push    eax ebx

        mov     ecx, [ebp + thread_data.datasocketnum]

        lea     edx, [ebp + thread_data.buffer]

        xor     edi, edi

        mcall   send

        pop     ebx ecx

        cmp     eax, -1

        je      socketerror                                     ; FIXME: not the correct error

;        cmp     eax, ecx

        cmp     ecx, BUFFERSIZE

        invoke  file.close, ebx

        invoke  con_write_asciiz, str2b

        mov     [ebp + thread_data.mode], MODE_NOTREADY

        sendFTP "226 Transfer OK, closing connection"

  .cannot_open:

        invoke  con_write_asciiz, str_notfound

        invoke  con_set_flags, 0x07

        sendFTP "550 No such file"

;

; Store a file on the server.

;

;------------------------------------------------

align 4

cmdSTOR:

        test    [ebp + thread_data.permissions], PERMISSION_WRITE

        ;sendFTP " Ready to receive"

        cmp     ecx, 1024 + 5

        jae     .cannot_open

        sub     ecx, 5

        call    open_datasock

        ; creat path

        dec     edi

        lea     esi, [ebp + thread_data.buffer + 5]

        mov     ecx, 1024

        cmp     byte [esi], '/'

        jne     .loop

        inc     esi

  .loop:

        lodsb

        cmp     al, 0x20

        jl      .done

        stosb

        loop    .loop

  .done:

        xor     al, al

        stosb

        lea     ebx, [ebp + thread_data.fpath]

        invoke  con_write_asciiz, str_newline

        ; open file

        test    eax, eax

        jz      .cannot_open

        push    eax

        pop     ebx

  .write_more:

        jnz     abort_transfer

        push    eax ebx

        mov     ecx, [ebp + thread_data.datasocketnum]

        lea     edx, [ebp + thread_data.buffer]

        xor     edi, edi

        mcall   recv

        pop     ebx ecx

        cmp     eax, -1

        je      socketerror                                     ; FIXME: not the correct error

        test    eax, eax

        push    edx

        lea     eax, [ebp + thread_data.buffer]                 ; FIXME: use another buffer!! if we receive something on control connection now, we screw up!

        invoke  file.write, ebx, eax, edx

        pop     edx

        cmp     eax, -1

        invoke  con_write_asciiz, str2

;        cmp     eax, ecx

        ;cmp     ecx, BUFFERSIZE