kolibrios/kernel/branches/net/applications/ftpd/commands.inc

align 4
parse_cmd:                              ; esi must point to command

        cmp     byte [esi+3], 0x20
        jae     @f
        mov     byte [esi+3], 0
       @@:

        mov     eax, [esi]
        and     eax, not 0x20202020     ; convert to upper case
        mov     edi, commands           ; list of commands to scan
  .scanloop:
        cmp     eax, [edi]
        jne     .try_next

        jmp     dword [edi+4]

  .try_next:
        add     edi, 8
        cmp     byte [edi], 0
        jne     .scanloop

  .error:
        mcall   send, [socketnum2], str500, str500.length, 0

        ret


align 4
commands:               ; all commands must be in uppercase

        db 'ABOR'
        dd cmdABOR
        db 'CDUP'
        dd cmdCDUP
        db 'CWD', 0
        dd cmdCWD
        db 'DELE'
        dd cmdDELE
        db 'LIST'
        dd cmdLIST
        db 'NLST'
        dd cmdNLST
        db 'NOOP'
        dd cmdNOOP
        db 'PASS'
        dd cmdPASS
        db 'PASV'
        dd cmdPASV
        db 'PWD', 0
        dd cmdPWD
        db 'PORT'
        dd cmdPORT
        db 'QUIT'
        dd cmdQUIT
        db 'RETR'
        dd cmdRETR
        db 'STOR'
        dd cmdSTOR
        db 'SYST'
        dd cmdSYST
        db 'TYPE'
        dd cmdTYPE
        db 'USER'
        dd cmdUSER
        db 'XPWD'
        dd cmdPWD
        db 0            ; end marker


align 4
cmdABOR:

        ret

align 4
cmdCDUP:

        cmp     byte [work_dir+1], 0
        je      .done

        mov     ecx, 1024
        xor     al, al
        mov     edi, work_dir+1024
        repne   scasb
        std
        dec     edi
        mov     al,'/'
        scasb
        cld
        mov     byte[edi], 0

  .done:
        mcall   send, [socketnum2], str250, str250.length, 0    ; command successful


        ret

align 4
cmdCWD:                 ; Change Working Directory

        sub     ecx, 4
        jb      .err
        add     esi, 4

  .scan:
        mov     edi, work_dir + 1

        cmp     byte [esi], '/'
        jne     @f
        inc     esi
        dec     ecx
        jz      .done
       @@:

  .loop:
        lodsb
        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

        mcall   send, [socketnum2], str250, str250.length, 0

        ret

  .up:
        lodsb
        cmp     al, '.'
        jne     .continue

        call    cmdCDUP
        jmp     .scan

  .err:

        ret

align 4
cmdDELE:

        ret

align 4
cmdLIST:

; If we are in active mode, it's time to open a data socket..
        cmp     [mode], MODE_ACTIVE
        jne     @f
        mcall   connect, [datasocketnum], datasock, datasock.length
        cmp     eax, -1
        je      socketerror
        mov     [datasocketnum], eax
  @@:

; Create fpath from home_dir and work_dir
        call    create_path

        push    fpath
        call    [con_write_asciiz]
        push    str_newline
        call    [con_write_asciiz]

; Start the search
        push    FA_ANY
        push    str_mask
        push    fpath
        call    [file.find.first]

        mov     edi, buffer
  .parse_file:
        test    eax, eax        ; did we find a file?
        jz      .done

        mov     edx, eax        ; yes, save the descripter

; first, convert the attributes
        test    [edx + FileInfoA.Attributes], FA_FOLDER
        jnz     .folder

        test    [edx + FileInfoA.Attributes], FA_READONLY
        jnz     .readonly

        mov     eax, '-rw-'
        stosd
        jmp     .attr

  .folder:
        mov     eax, 'drwx'
        stosd
        jmp     .attr

  .readonly:
        mov     eax, '-r--'
        stosd

  .attr:
        mov     eax, 'rw-r'
        stosd
        mov     ax, 'w-'
        stosw
        mov     al, ' '
        stosb

; now..
        mov     ax, '1 '
        stosw

; now write owner, everything is owned by FTP, woohoo!
        mov     eax, 'FTP '
        stosd
        stosd

; now the filesize in ascii
        mov     ebx, [edx + FileInfoA.FileSizeLow]
        call    dword_to_ascii

        mov     al, ' '
        stosb

; then date (month/day/year)
        movzx   ebx, [edx + FileInfoA.DateModify + FileDateTime.month]
        mov     eax, [months + 4*ebx]
        stosd

        movzx   ebx, [edx + FileInfoA.DateModify + FileDateTime.day]
        call    dword_to_ascii

        mov     al, ' '
        stosb

        movzx   ebx, [edx + FileInfoA.DateModify + FileDateTime.year]
        call    dword_to_ascii

        mov     al, ' '
        stosb

; and last but not least, filename
        lea     esi, [edx + FileInfoA.FileName]
        mov     ecx, 250
  .nameloop:
        lodsb
        test    al, al
        jz      .namedone
        stosb
        loop    .nameloop

; insert a cr lf
  .namedone:
        mov     ax, 0x0a0d
        stosw

; check next file
        push    edx
        call    [file.find.next]
        jmp     .parse_file

; close file desc
  .done:
        push    edx
        call    [file.find.close]

; append the string with a 0
        xor     al, al
        stosb

; Warn the client we're about to send the data
        mcall   send, [socketnum2], str150, str150.length, 0    ; here it comes..

; and send it to the client
        lea     esi, [edi - buffer]
        mcall   send, [datasocketnum], buffer, , 0

; close the data socket..
        mcall   close, [datasocketnum]

        cmp     [mode], MODE_PASSIVE_OK
        jne     @f
        mov     [mode], MODE_PASSIVE_WAIT
      @@:

; And send "transfer ok" on the base connection
        mcall   send, [socketnum2], str226, str226.length, 0

        ret


align 4
cmdNLST:

        ret

align 4
cmdNOOP:

        ret

align 4
cmdPASS:

        mcall   send, [socketnum2], str230, str230.length, 0

        push    str_pass_ok
        call    [con_write_asciiz]

        mov     [state], STATE_ACTIVE

        ret

align 4
cmdPASV:

; 227 Entering Passive Mode (a1,a2,a3,a4,p1,p2)
; where a1.a2.a3.a4 is the IP address and p1*256+p2 is the port number.

        mcall   socket, AF_INET4, SOCK_STREAM, 0
        cmp     eax, -1
;        je      .err
        mov     [passivesocknum], eax

        mov     [datasock.port], 2000
        mov     [datasock.ip], 0

        mcall   bind, [passivesocknum], datasock, datasock.length
        cmp     eax, -1
        je      bind_err

        mcall   listen, [passivesocknum], 1

        mov     [mode], MODE_PASSIVE_WAIT

        mov     edi, buffer
        mov     eax, '227 '     ; FIXME (now hardcoded to 127.0.0.1:2000)
        stosd
        mov     eax, '(127'
        stosd
        mov     eax, ',0,0'
        stosd
        mov     eax, ',1,7'
        stosd
        mov     eax, ',208'
        stosd
        mov     al, ')'
        stosb
        mov     ax, 0x0a0d
        stosw
        xor     al, al
        stosb

        lea     esi, [edi - buffer]

        mcall   send, [socketnum2], buffer, ,0

        ret

align 4
cmdPWD:         ; Print Working Directory

        mov     dword[buffer], '257 '
        mov     byte[buffer+4], '"'

        lea     edi, [buffer+5]
        mov     esi, work_dir
        mov     ecx, 1024
  .loop:
        lodsb
        or      al, al
        jz      .ok
        stosb
        dec     ecx
        jnz     .loop

  .ok:
        mov     dword[edi], '"' + 0x000a0d00    ; '"',13,10,0
        lea     esi, [edi - buffer + 4]

        mcall   send, [socketnum2], buffer, , 0

;        push    work_dir
;        push    str_pwd
;        call    [con_printf]

        ret

align 4
cmdPORT:

; PORT a1,a2,a3,a4,p1,p2
; IP address a1.a2.a3.a4, port p1*256+p2

        mov     [mode], MODE_ACTIVE

        lea     esi, [esi+5]
        xor     edx, edx

        call    ascii_to_byte
        mov     dh, bl
        inc     esi
        call    ascii_to_byte
        mov     dl, bl
        shl     edx, 16
        inc     esi
        call    ascii_to_byte
        mov     dh, bl
        inc     esi
        call    ascii_to_byte
        mov     dl, bl
        inc     esi

        mov     [datasock.ip], edx

        call    ascii_to_byte
        mov     dh, bl
        inc     esi
        call    ascii_to_byte
        mov     dl, bl

        mov     [datasock.port], dx

        mcall   socket, AF_INET4, SOCK_STREAM, 0
        cmp     eax, -1
        je      .err
        mov     [datasocketnum], eax

        mcall   send, [socketnum2], str225, str225.length, 0
        ret

  .err:

        mcall   send, [socketnum2], str425, str425.length, 0
        ret

align 4
cmdQUIT:

        mcall   send, [socketnum2], str221, str221.length, 0
        mcall   close, [socketnum2]

        ret

align 4
cmdRETR:

        sub     ecx, 5
        jb      .cannot_open

        cmp     [mode], MODE_ACTIVE
        jne     @f
        push    esi
        mcall   connect, [datasocketnum], datasock, datasock.length
        pop     esi
        cmp     eax, -1
        je      socketerror
        mov     [datasocketnum], eax
  @@:

        push    esi
        call    create_path
        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

        push    fpath
        call    [con_write_asciiz]
        push    str_newline
        call    [con_write_asciiz]

        push    O_READ
        push    fpath
        call    [file.open]
        test    eax, eax
        jz      .cannot_open

        push    eax
        mcall   send, [socketnum2], str150, str150.length, 0    ; here it comes..
        pop     ebx

  .read_more:
        push    BUFFERSIZE
        push    buffer
        push    ebx
        call    [file.read]
        cmp     eax, -1
        je      .cannot_open    ; fixme: this is not the correct error

        push    eax
        push    ebx
        mov     esi, eax
        mcall   send, [datasocketnum], buffer, , 0
        pop     ebx
        pop     ecx
        cmp     eax, -1
        je      socketerror

        cmp     ecx, BUFFERSIZE
        je      .read_more

        mcall   close, [datasocketnum]

        cmp     [mode], MODE_PASSIVE_OK
        jne     @f
        mov     [mode], MODE_PASSIVE_WAIT
      @@:

        mcall   send, [socketnum2], str226, str226.length, 0    ; transfer ok

        ret

  .cannot_open:

        pushd   0x0c
        call    [con_set_flags]

        push    str_notfound
        call    [con_write_asciiz]

        pushd   0x07
        call    [con_set_flags]

        mcall   send, [socketnum2], str550, str550.length, 0    ; file not found

        ret

align 4
cmdSTOR:

        ret

align 4
cmdSYST:

        mcall   send, [socketnum2], str215, str215.length, 0

        ret

align 4
cmdTYPE:


        cmp     ecx, 6
        jb      parse_cmd.error

        mov     al, byte[esi+5]
        and     al, not 0x20

        cmp     al, 'A'
        je      .ascii
        cmp     al, 'E'
        je      .ebdic
        cmp     al, 'I'
        je      .image
        cmp     al, 'L'
        je      .local

        jmp     parse_cmd.error

  .ascii:
        mov     [type], TYPE_ASCII
        jmp     .subtype

  .ebdic:
        mov     [type], TYPE_EBDIC

  .subtype:

        cmp     ecx, 8
        jb      .non_print

        mov     al, byte[esi+7]
        and     al, not 0x20

        cmp     al, 'N'
        je      .non_print
        cmp     al, 'T'
        je      .telnet
        cmp     al, 'C'
        je      .asacc

        jmp     parse_cmd.error

  .non_print:
        or      [type], TYPE_NP
        jmp     .ok

  .telnet:
        or      [type], TYPE_TELNET
        jmp     .ok

  .asacc:
        or      [type], TYPE_ASA
        jmp     .ok

  .image:
        mov     [type], TYPE_IMAGE
        jmp     .ok

  .local:
        cmp     ecx, 8
        jb      parse_cmd.error

        mov     al, byte[esi+7]
        sub     al, '0'
        jb      parse_cmd.error
        cmp     al, 9
        ja      parse_cmd.error
        or      al, TYPE_LOCAL
        mov     [type], al

  .ok:
        mcall   send, [socketnum2], str200, str200.length, 0

        ret

align 4
cmdUSER:

        mcall   send, [socketnum2], str331, str331.length, 0
        mov     [state], STATE_LOGIN

        mov     byte [work_dir], "/"
        mov     byte [work_dir+1], 0

        push    str_logged_in
        call    [con_write_asciiz]

        ret







align 4         ; esi = ptr to str
ascii_to_byte:

        xor     ebx, ebx

  .loop:

        movzx   eax, byte[esi]
        sub     al, '0'
        jb      .done
        cmp     al, 9
        ja      .done
        lea     ebx, [ebx*4 + ebx]
        shl     ebx, 1
        add     ebx, eax
        inc     esi

        jmp     .loop

  .done:
        ret

align 4
dword_to_ascii: ; edi = ptr where to write, ebx is number

        mov     eax, '1'
        stosb

        ret

align 4
create_path:            ; combine home_dir and work_dir strings into fpath
        mov     edi, fpath
        mov     esi, home_dir
        mov     ecx, 1024

  .loop1:
        lodsb
        or      al, al
        jz      .next
        stosb
        loop    .loop1
  .next:

        cmp     byte[edi-1], '/'
        jne     @f
        dec     edi
       @@:

        mov     esi, work_dir
        mov     ecx, 1024

  .loop2:
        lodsb
        or      al, al
        jz      .done
        stosb
        loop    .loop2

  .done:
        stosb

        ret


align 4
socketerror:

        pushd   0x0c
        call    [con_set_flags]

        push    str_sockerr
        call    [con_write_asciiz]

        pushd   0x07
        call    [con_set_flags]

        mcall   send, [socketnum2], str425, str425.length, 0    ; data connection error

        ret




str150  db '150 Here it comes...', 13, 10
.length = $ - str150
str200  db '200 Command OK.', 13, 10
.length = $ - str200
str215  db '215 UNIX type: L8', 13, 10
.length = $ - str215
str220  db '220 KolibriOS FTP Daemon 1.0', 13, 10
.length = $ - str220
str221  db '221 Bye!', 13, 10
.length = $ - str221
str225  db '225 Data connection open', 13, 10
.length = $ - str225
str226  db '226 Transfer OK, Closing connection', 13, 10
.length = $ - str226
str230  db '230 You are now logged in.', 13, 10
.length = $ - str230
str250  db '250 command successful', 13, 10
.length = $ - str250
str331  db '331 Please specify the password.', 13, 10
.length = $ - str331
str421  db '421 Timeout!', 13, 10
.length = $ - str421
str425  db '425 Cant open data connection.', 13, 10
.length = $ - str425
str500  db '500 Unsupported command', 13, 10
.length = $ - str500
str550  db '550 No such file', 13, 10
.length = $ - str550