Subversion Repositories Kolibri OS

        stack           rb 0

        home_dir        rb 1024

        fpath           rb 1024*3       ; 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 ?

        buffer_ptr      dd ?

        datasock        sockaddr_in

        buffer          rb BUFFERSIZE

        xor     edi, edi

        mcall   send, [ebp + thread_data.socketnum], .string, .length

        jmp     @f

.string db str, 13, 10

.length = $ - .string

@@:

}

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

;

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

        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

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

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

;

; 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 "/" ?

        mov     ecx, 1024

        lea     edi, [ebp + thread_data.work_dir]

        repne   scasb

        sub     edi, 3

        mov     al,'/'

        std

        neg     ecx

        add     ecx, 1024

        repne   scasb

        cld

        mov     byte[edi+1], 0

  .done:

        lea     eax, [ebp + thread_data.work_dir]

        push    eax

        call    [con_write_asciiz]

        push    str_newline

        call    [con_write_asciiz]

        sendFTP "250 Command succesul"

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

;

; Change Working Directory.

;

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

align 4

cmdCWD:

        test    [ebp + thread_data.permissions], PERMISSION_CD

        sub     ecx, 4

        add     esi, 4

        mov     ecx, 1024

        cmp     byte [esi], '/'

        je      .loop

  .scan:

        push    ecx

        mov     ecx, 1024

  .find_zero:

        cmp     byte [edi], 0

        je      .found_zero

        inc     edi

        loop    .find_zero

  .found_zero:

        pop     ecx

  .scan2:

        cmp     byte [esi], '/'

        inc     esi

        dec     ecx

        jz      .done

       @@:

  .loop:

        cmp     al, 0x20

        jb      .done

        cmp     al, '.'

        je      .up

  .continue:

        stosb

        loop    .loop

  .done:

        cmp     byte [edi-1], '/'

        je      @f

        mov     byte [edi], '/'

        inc     edi

       @@:

        mov     byte [edi], 0

; Print the new working dir on the console

        push    eax

        call    [con_write_asciiz]

        push    str_newline

        call    [con_write_asciiz]

        sendFTP "250 Command succesful"

  .up:

        cmp     al, '.'

        jne     .continue

;;;;        TODO: find second last '\' in work_dir and make next char zero

        jmp     .scan2

  .err:

        ret

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

;

; Delete a file from the server.

;

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

align 4

cmdDELE:

        test    [ebp + thread_data.permissions], PERMISSION_DELETE

        ret

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

;

; List the files in the current working directory.

;

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

align 4

cmdLIST:

        test    [ebp + thread_data.permissions], PERMISSION_EXEC

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

        jne     @f

        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

  @@:

; 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 + FileDateTime.day]

        mov     al, ' '

        stosb

        movzx   eax, [ebx + FileInfoA.DateModify + FileDateTime.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

;;;        mov     eax, ebx        ;;;;;

; close file desc

        invoke  file.find.close, ebx                            ; ebx is the

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

        sub     esi, edx

        xor     edi, edi

        mcall   send

; close the data socket..

        mcall   close, [ebp + thread_data.datasocketnum]

        sendFTP "226 Transfer 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:

        lea     esi, [esi + 5]

; 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

        jnz     .incorrect

        cmp     dword [edi], -1

        je      .incorrect

        cmp     byte[edi], 0

        je      .pass_ok

; compare with received password

        cmp     byte [esi], 0x20

        jae     .incorrect

        cmp     byte [edi], 0

        jne     .incorrect

  .pass_ok:

        mov     [ebp + thread_data.permissions], eax

        invoke  con_write_asciiz, str_pass_ok

        sendFTP "230 You are now logged in"

        ret

  .2:

        mov     [ebp + thread_data.state], STATE_CONNECTED

        sendFTP "530 Login incorrect"

        ret

align 4

        sendFTP "503 Login with USER first"

        ret

align 4

        sendFTP "230 Already logged in"

        ret

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

;

; Initiate a passive dataconnection.

;

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

align 4

cmdPASV:

; Open a new TCP socket

        cmp     eax, -1

        je      socketerror

        mov     [ebp + thread_data.passivesocknum], eax

; Bind it to a known local port

        pushw   [pasvport]

        popw    [ebp + thread_data.datasock.sin_port]

        inc     [pasvport]

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

        mov     ecx, eax                                        ; passivesocketnum

        mov     esi, sizeof.thread_data.datasock

        mcall   bind

        cmp     eax, -1

;        je      bind_err       ; TODO

; And set it to listen!

        cmp     eax, -1

;        je      listen_err     ; TODO

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

        mov     eax, '227 '     ; FIXME (now hardcoded to 127.0.0.1:2000)

        stosd

        mov     al, '('

        stosb

; ip

        mov     eax, 127

        call    dword_to_ascii

        mov     al, ','

        stosb

        mov     eax, 0

        call    dword_to_ascii

        mov     al, ','

        stosb

        mov     eax, 0

        call    dword_to_ascii

        mov     al, ','

        stosb

        mov     eax, 1

        call    dword_to_ascii

        mov     al, ','

        stosb

; port

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

        call    dword_to_ascii

        mov     al, ','

        stosb

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

        call    dword_to_ascii

; ')', 13, 10, 0

        mov     eax, ')' + 0x000a0d00

        stosd

        lea     esi, [edi - thread_data.buffer]

        mov     ecx, [ebp + thread_data.socketnum]

        lea     edx, [ebp + thread_data.buffer]

        xor     edi, edi

        mcall   send

        ret

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

;

; 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 + 4]

        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     bh, al

        inc     esi

        call    ascii_to_byte

        mov     bl, al

; Save it in datasock too

; 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

        cmp     [ebp + thread_data.mode], MODE_ACTIVE

        push    ecx esi

        mov     ecx, [ebp + thread_data.datasocketnum]

        lea     edx, [ebp + thread_data.datasock]

        mov     esi, sizeof.thread_data.datasock

        mcall   connect

        pop     esi ecx

        cmp     eax, -1

        je      socketerror

  @@:

        push    ecx esi

        pop     esi ecx

        dec     edi

        add     esi, 5

  .loop:

        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

        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

        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

        mov     edx, [ebp]

        ret

;

; Store a file on the server.

;

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

align 4

cmdSTOR:

        test    [ebp + thread_data.permissions], PERMISSION_WRITE

        jnz     abort_transfer

;;;;

        ret

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

;

; Send information about the system.

;

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

align 4

cmdSYST:

        sendFTP "215 UNIX type: L8"

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

;

; Choose the file transfer type.

;

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

align 4

cmdTYPE:

        cmp     ecx, 6

        mov     al, byte[esi+5]

        cmp     al, 'A'

        cmp     al, 'E'

        je      .ebdic

        cmp     al, 'I'

        je      .image

        cmp     al, 'L'

        je      .local

        jmp     parse_cmd.error

  .ascii:

        jmp     .subtype

  .ebdic:

  .subtype:

        jb      .non_print

        mov     al, byte[esi+7]

        cmp     al, 'N'

        cmp     al, 'T'

        je      .telnet

        cmp     al, 'C'

        je      .asacc

        jmp     parse_cmd.error

  .non_print:

        jmp     .ok

  .telnet:

        jmp     .ok

  .asacc:

        jmp     .ok

  .image:

        jmp     .ok

  .local:

        jb      parse_cmd.error

        mov     al, byte[esi+7]

        jb      parse_cmd.error                         ; FIXME: this is not the correct errormessage

        cmp     al, 9

        ja      parse_cmd.error                         ; FIXME

        or      al, TYPE_LOCAL

        mov     [ebp + thread_data.type], al

  .ok:

        ret

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

;

; Login to the server, step one of two.                         ;;; TODO: prevent buffer overflow!

;

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

align 4

cmdUSER:

        lea     esi, [esi + 5]

  .loop:

        lodsb

        stosb

        cmp     al, 0x20

        jae     .loop

        mov     byte [edi-1], 0