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, [edx + 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

;        edx = 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, 4+4*4

        jne     .scanloop

  .error:

        jb      login_first

        sendFTP "500 Unsupported command"

        ret

  .got_it:

        jmp     dword [edi + 4 + eax]

        dd 'ABOR'

;        dd 'ACCT

;        dd login_fitst, login_first, login_first, cmd_ACCT

;        dd 'APPE'

;        dd login_fitst, login_first, login_first, cmd_APPE

        dd 'CDUP'

        dd login_first, login_first, login_first, cmdCDUP

        dd 'CWD'

        dd login_first, login_first, login_first, cmdCWD

        dd 'DELE'

        dd login_first, login_first, login_first, cmdDELE

;        dd 'HELP'

;        dd login_fitst, login_first, login_first, cmd_HELP

        dd 'LIST'

        dd login_first, login_first, login_first, cmdLIST

;        dd 'MDTM'

;        dd login_fitst, login_first, login_first, cmd_MDTM

;        dd 'MKD'

;        dd login_fitst, login_first, login_first, cmd_MKD

;        dd 'MODE'

;        dd login_fitst, login_first, login_first, cmd_MODE

        dd 'NLST'

        dd login_first, login_first, login_first, cmdNLST

        dd 'NOOP'

        dd login_first, login_first, login_first, cmdNOOP

        dd 'PASS'

        dd cmdPASS.0,   cmdPASS    , cmdPASS.2,   cmdPASS.3

        dd 'PASV'

        dd login_first, login_first, login_first, cmdPASV

        dd 'PORT'

        dd login_first, login_first, login_first, cmdPORT

        dd 'PWD'

        dd login_first, login_first, login_first, cmdPWD

        dd 'QUIT'

        dd cmdQUIT,     cmdQUIT,     cmdQUIT,     cmdQUIT

;        dd 'REIN'

;        dd login_fitst, login_first, login_first, cmd_REIN

;        dd 'REST'

;        dd login_fitst, login_first, login_first, cmd_REST

        dd 'RETR'

        dd login_first, login_first, login_first, cmdRETR

;        dd 'RMD'

;        dd login_fitst, login_first, login_first, cmd_RMD

;        dd 'RNFR'

;        dd login_fitst, login_first, login_first, cmd_RNFR

;        dd 'RNTO'

;        dd login_fitst, login_first, login_first, cmd_RNTO

;        dd 'SITE'

;        dd login_fitst, login_first, login_first, cmd_SITE

;        dd 'SIZE'

;        dd login_fitst, login_first, login_first, cmd_SIZE

;        dd 'STAT'

;        dd login_fitst, login_first, login_first, cmd_STAT

        dd 'STOR'

        dd login_first, login_first, login_first, cmdSTOR

;        dd 'STOU'

;        dd login_fitst, login_first, login_first, cmd_STOU

;        dd 'STRU'

;        dd login_fitst, login_first, login_first, cmd_STRU

        dd 'SYST'

        dd login_first, login_first, login_first, cmdSYST

        dd 'TYPE'

        dd login_first, login_first, login_first, cmdTYPE

        dd 'USER'

        dd 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

        pushd   0x0c

        call    [con_set_flags]

        push    str_sockerr

        call    [con_write_asciiz]

        pushd   0x07

        call    [con_set_flags]

        mov     edx, [ebp]

        ret

align 4

        and     [edx + thread_data.permissions], not ABORT

        mov     [edx + thread_data.mode], MODE_NOTREADY

        push    ebx

        call    [file.close]

        mcall   close, [edx + thread_data.datasocketnum]

        mov     edx, [ebp]

        sendFTP "530 Transfer aborted"

        ret

align 4

        call    ascii_to_byte

        cmp     byte [esi], cl

        jne     .err

        call    ascii_to_byte

        cmp     byte [esi], cl

        jne     .err

        shl     ebx, 16

        call    ascii_to_byte

        cmp     byte [esi], cl

        jne     .err

        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

        mov     edx, [ebp]

        lea     esi, [edx + thread_data.home_dir]

        mov     ecx, 1024

  .loop1:

        lodsb

        or      al, al

        jz      .next

        stosb

        loop    .loop1

  .next:

        cmp     byte[edi-1], '/'

        dec     edi

       @@:

        lea     esi, [edx + thread_data.work_dir]

  .loop2:

        lodsb

        or      al, al

        jz      .done

        stosb

        loop    .loop2

  .done:

        ret

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

;

; This command aborts the current filetransfer.

;

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

align 4

cmdABOR:

        or      [edx + thread_data.permissions], ABORT

        ret

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

;

; Change the directory to move up one level.

;

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

align 4

cmdCDUP:

        test    [edx + thread_data.permissions], PERMISSION_CD

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

        mov     ecx, 1024

        lea     edi, [edx + thread_data.work_dir]

        repne   scasb

        std

        dec     edi

        dec     edi

        dec     edi

        mov     al,'/'

        repne   scasb

        cld

        mov     byte[edi+1], 0

  .done:

        lea     eax, [edx + 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    [edx + thread_data.permissions], PERMISSION_CD

        sub     ecx, 4

        add     esi, 4

  .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    [edx + thread_data.permissions], PERMISSION_DELETE

        ret

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

;

; List the files in the current working directory.

;

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

align 4

cmdLIST:

        test    [edx + thread_data.permissions], PERMISSION_EXEC

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

        jne     @f

        mov     ecx, [edx + thread_data.datasocketnum]

        lea     edx, [edx + 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     eax, [edx + thread_data.fpath]

        call    [con_write_asciiz]

        push    str_newline

        call    [con_write_asciiz]

; Start the search

        push    FA_ANY

        push    str_mask

        lea     eax, [edx + thread_data.fpath]

        push    eax

        call    [file.find.first]

        test    eax, eax

        mov     edx, [ebp]

  .parse_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, ' '

; then date (month/day/year)

        mov     eax, [months + 4*eax]

        stosd

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

        mov     al, ' '

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

        mov     al, ' '

; 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    [edx + thread_data.permissions], ABORT

; check next file

        call    [file.find.next]

        jmp     .parse_file

; close file desc

        push    eax                             ; file discriptor is still in eax at this point!

        call    [file.find.close]

; append the string with a 0

        stosb

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

        push    edi

        sendFTP "150 Here it comes.."

        pop     esi

; and send it to the client

        mov     ecx, [edx + thread_data.datasocketnum]

        lea     edx, [edx + thread_data.buffer]

        sub     esi, edx

        xor     edi, edi

        mcall   send

; close the data socket..

        mcall   close, [edx + thread_data.datasocketnum]

        mov     [edx + thread_data.mode], MODE_NOTREADY

        sendFTP "226 Transfer OK"

  .nosuchdir:

        ret

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

;

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

;

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

align 4

cmdNLST:

        test    [edx + thread_data.permissions], PERMISSION_EXEC

        ; TODO: same as list but simpler output format

        ret

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

;

; No operation, just keep the connection alive.

;

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

align 4

cmdNOOP:

        ret

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

;

; Second phase of login process, client provides password.

;

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

align 4

cmdPASS:

        lea     esi, [esi + 5]

        lea     edi, [edx + thread_data.buffer + 512]           ; temp pass

        lea     eax, [edx + thread_data.fpath]                  ; temp username

        invoke  ini.get_str, path2, eax, str_pass, edi, 512

        test    eax, eax

        jnz     .incorrect

        repe    cmpsb

        cmp     byte [esi], 0x20

        cmp     byte [edi], 0

  .pass_ok:

        invoke  ini.get_int, path2, eax, str_mode, 0

        mov     [edx + thread_data.permissions], eax

        push    str_pass_ok

        mov     edx, [ebp]                                      ; thread_data pointer

        sendFTP "230 You are now logged in"

        ret

  .2:

        mov     [edx + 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     edx, [ebp]                                      ; thread_data pointer

        mov     [edx + thread_data.passivesocknum], eax

; Bind it to a known local port

        mov     [edx + thread_data.datasock.sin_port], 2000

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

        mov     ecx, eax                                        ; passivesocketnum

        mov     esi, sizeof.thread_data.datasock

        mcall   bind

        cmp     eax, -1

;        je      bind_err

; And set it to listen!

        cmp     eax, -1

;        je      listen_err

; Tell our thread we are ready to accept incoming calls

        mov     [edx + thread_data.mode], MODE_PASSIVE_WAIT

; 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

        mov     eax, 7

        call    dword_to_ascii

        mov     al, ','

        stosb

        mov     eax, 208

        call    dword_to_ascii

; ')', 13, 10, 0

        mov     eax, ')' + 0x000a0d00

        stosd

        lea     esi, [edi - thread_data.buffer]

        mov     ecx, [edx + thread_data.socketnum]

        lea     edx, [edx + thread_data.buffer]

        xor     esi, esi

        mcall   send

        ret

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

;

; Print the current working directory.

;

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

align 4

cmdPWD:

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

        lea     edi, [edx + 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, edx

        mov     ecx, [edx + thread_data.socketnum]

        lea     edx, [edx + thread_data.buffer]

        xor     edi, edi

        mcall   send

        mov     edx, [ebp]

        lea     eax, [edx + thread_data.work_dir]

        push    eax

        call    [con_write_asciiz]

        push    str_newline

        call    [con_write_asciiz]

        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     edx, [ebp]

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

; Now the same with portnumber

        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     edx, [ebp]                                      ; thread_data pointer

        mov     [edx + thread_data.mode], MODE_ACTIVE

        sendFTP "225 Data connection open"

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

;

; Close the connection with client.

;

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

align 4

cmdQUIT:

        mcall   close, [edx + thread_data.datasocketnum]

        sendFTP "221 Bye!"

        add     esp, 4          ; get rid of call return address

;

; Retrieve a file from the ftp server.

;

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

align 4

cmdRETR:

        test    [edx + thread_data.permissions], PERMISSION_READ

        sub     ecx, 5

        cmp     [edx + thread_data.mode], MODE_ACTIVE

        push    esi

        mov     ecx, [edx + thread_data.datasocketnum]

        lea     edx, [edx + thread_data.datasock]

        mov     esi, sizeof.thread_data.datasock

        mcall   connect

        pop     esi

        cmp     eax, -1

        je      socketerror

  @@:

        push    esi

        pop     esi

        dec     edi

        add     esi, 5

        mov     ecx, 1024

  .loop:

        lodsb

        cmp     al, 0x20

        jl      .done

        stosb

        loop    .loop

  .done:

        xor     al, al

        stosb

        lea     eax, [edx + thread_data.fpath]

        call    [con_write_asciiz]

        push    str_newline

        call    [con_write_asciiz]

        mov     edx, [ebp]

        lea     eax, [edx + thread_data.fpath]

        push    eax

        call    [file.open]

        test    eax, eax

        jz      .cannot_open

        mov     edx, [ebp]

        sendFTP "150 Here it comes.."

        pop     ebx

  .read_more:

        test    [edx + thread_data.permissions], ABORT

        jnz     abort_transfer

        push    BUFFERSIZE

        push    eax

        push    ebx

        call    [file.read]

        cmp     eax, -1

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

        mov     edx, [ebp]

        mov     esi, eax

        mov     ecx, [edx + thread_data.datasocketnum]

        lea     edx, [edx + thread_data.buffer]

        xor     esi, esi

        mcall   send

        pop     ebx ecx

        mov     edx, [ebp]                                      ; thread_data pointer

        cmp     eax, -1

        je      socketerror

        cmp     ecx, BUFFERSIZE

        mcall   close, [edx + thread_data.datasocketnum]

        push    ebx

        sendFTP "226 Transfer OK, closing connection"

  .cannot_open:

        call    [con_set_flags]

        push    str_notfound

        call    [con_write_asciiz]

        pushd   0x07

        call    [con_set_flags]

        mov     edx, [ebp]

        ret

;

; Store a file on the server.

;

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

align 4

cmdSTOR:

        test    [edx + 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