WebSVN – Kolibri OS – Diff – /kernel/branches/net/applications/ftpd/commands.inc


struct  thread_data
                        rb 1024
        stack           rb 0
 
        home_dir        rb 1024         ; home directory in wich the user is locked, asciiz
        work_dir        rb 1024         ; working directory, must at all times begin and end with a '/', 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/..
        mode            db ?    ; active/passive
        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
ends
 
;------------------------------------------------
 
 
 
 
 
 
 
 
 
 
 
 
; parse_cmd
;
; 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
        ja      .ok
        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]
        and     eax, not 0x20202020     ; convert to upper case
        mov     edi, commands           ; list of commands to scan
  .scanloop:
        cmp     eax, [edi]
        je      .got_it
 
        add     edi, 5*4
        cmp     byte [edi], 0
        jne     .scanloop
 
  .error:
        cmp     [ebp + thread_data.state], STATE_ACTIVE
        jb      login_first
        sendFTP "500 Unsupported command"
        ret
 
  .got_it:
        mov     eax, [ebp + thread_data.state]
        jmp     dword [edi + 4 + eax]
 
 
align 4
iglobal
commands:               ; all commands must be in uppercase
 
        dd 'ABOR', login_first, login_first, login_first, cmdABOR
;        dd 'ACCT', login_first, login_first, login_first, cmd_ACCT
;        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
login_first:
        sendFTP "530 Please login with USER and PASS"
        ret
 
align 4
permission_denied:
        sendFTP "550 Permission denied"
        ret
 
align 4
socketerror:
        invoke  con_set_flags, 0x0c
        invoke  con_write_asciiz, str_sockerr
        invoke  con_set_flags, 0x07
 
        sendFTP "425 Can't open data connection"
        ret
 
align 4
abort_transfer:
        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"
        ret
 
align 4
ip_to_dword:    ; esi = ptr to str, cl = separator ('.', ',')
 
        call    ascii_to_byte
        mov     bl, al
        cmp     byte [esi], cl
        jne     .err
        inc     esi
 
        call    ascii_to_byte
        mov     bh, al
        cmp     byte [esi], cl
        jne     .err
        inc     esi
 
        shl     ebx, 16
 
        call    ascii_to_byte
        mov     bl, al
        cmp     byte [esi], cl
        jne     .err
        inc     esi
 
        call    ascii_to_byte
        mov     bh, al
 
        ror     ebx, 16
        ret
 
  .err:
        xor     ebx, ebx
        ret
 
align 4         ; esi = ptr to str, output in eax
ascii_to_byte:
 
        xor     eax, eax
        push    ebx
 
  .loop:
        movzx   ebx, byte[esi]
        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:
        pop     ebx
        ret
 
align 4
dword_to_ascii: ; edi = ptr where to write, eax is number
 
        push    edx ebx ecx
        mov     ebx, 10
        xor     ecx, ecx
 
       @@:
        xor     edx, edx
        div     ebx
        add     edx, '0'
        pushw   dx
        inc     ecx
        test    eax, eax
        jnz     @r
 
       @@:
        popw    ax
        stosb
        dec     ecx
        jnz     @r
 
        pop     ecx ebx edx
        ret
 
align 4
create_path:            ; combine home_dir and work_dir strings into fpath
 
        lea     edi, [ebp + thread_data.fpath]
        lea     esi, [ebp + thread_data.home_dir]
        mov     ecx, 1024
  .loop1:
        lodsb
        cmp     al, 0x20
        jb      .next
        stosb
        loop    .loop1
  .next:
 
        cmp     byte[edi-1], '/'
        jne     @f
        dec     edi
       @@:
 
        lea     esi, [ebp + thread_data.work_dir]
        mov     ecx, 1024
  .loop2:
        lodsb
        cmp     al, 0x20
        jb      .done
        stosb
        loop    .loop2
 
  .done:
        xor     al, al
        stosb
        ret
 
 
align 4
nextpasvport:
 
        inc     [pasv_port]
 
        mov     ax, [pasv_port]
        cmp     ax, [pasv_start]
        jb      .restart
        cmp     ax, [pasv_end]
        ja      .restart
 
        ret
 
  .restart:
        pushw   [pasv_start]
        popw    [pasv_port]
 
        ret
 
 
;------------------------------------------------
; "ABOR"
;
; This command aborts the current filetransfer.
;
;------------------------------------------------
align 4
cmdABOR:
 
        or      [ebp + thread_data.permissions], ABORT
        sendFTP "250 Command succesul"
        ret
 
;------------------------------------------------
; "CDUP"
;
; Change the directory to move up one level.
;
;------------------------------------------------
align 4
cmdCDUP:
 
        test    [ebp + thread_data.permissions], PERMISSION_CD
        jz      permission_denied
 
        cmp     byte [ebp + thread_data.work_dir+1], 0  ; are we in "/" ?
        je      .done                                   ; if so, we cant go up..
 
; find the end of asciiz string work_dir
        mov     ecx, 1024
        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:
; Print the new working dir on the console
        lea     eax, [ebp + thread_data.work_dir]
        invoke  con_write_asciiz, eax
        invoke  con_write_asciiz, str_newline
 
        sendFTP "250 Command succesul"
        ret
 
;------------------------------------------------
; "CWD"
;
; Change Working Directory.
;
;------------------------------------------------
align 4
cmdCWD:
 
        test    [ebp + thread_data.permissions], PERMISSION_CD
        jz      permission_denied
 
; do we have enough parameters?
        sub     ecx, 4
        jbe     .err
 
; get ready to copy the path
        add     esi, 4
        mov     ecx, 1024
        lea     edi, [ebp + thread_data.work_dir]
 
; if received dir starts with '/', we will simply copy it
; If not, we will append the current path with the received path.
        cmp     byte [esi], '/'
        je      .copyloop
 
; Find the end of work_dir string.
        xor     al, al
  .find_zero:
        repne   scasb
        dec     edi
 
; and now append work_dir with received string
        mov     ecx, 1024
 
; scan for end byte, or '.'
  .copyloop:
        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
  .done:
        cmp     byte [edi-1], '/'
        je      @f
        mov     byte [edi], '/'
        inc     edi
       @@:
        mov     byte [edi], 0
 
; Print the new working dir on the console
        lea     eax, [ebp + thread_data.work_dir]
        invoke  con_write_asciiz, eax
        invoke  con_write_asciiz, str_newline
 
        sendFTP "250 Command succesful"
        ret
 
  .err:
        sendFTP "550 Directory does not exist"
        ret
 
;------------------------------------------------
; "DELE"
;
; Delete a file from the server.
;
;------------------------------------------------
align 4
cmdDELE:
 
        test    [ebp + thread_data.permissions], PERMISSION_DELETE
        jz      permission_denied
 
        ret
 
;------------------------------------------------
; "LIST"
;
; List the files in the current working directory.
;
;------------------------------------------------
align 4
cmdLIST:
 
        test    [ebp + thread_data.permissions], PERMISSION_EXEC
        jz      permission_denied
 
        cmp     [ebp + thread_data.mode], MODE_PASSIVE_OK
        je      .start
 
; If we are in active mode, it's time to open a data socket..
        cmp     [ebp + thread_data.mode], MODE_ACTIVE
        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
        jmp     .start
 
; If we are still in passive_wait, it's time we accept an incomming call..
  .not_active:
        cmp     [ebp + thread_data.mode], MODE_PASSIVE_WAIT
        jne     socketerror
  .try_now:
        mov     ecx, [ebp + thread_data.passivesocknum]
        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
 
  .start:
; Create fpath from home_dir and work_dir
        call    create_path
 
        lea     ebx, [ebp + thread_data.fpath]
        invoke  con_write_asciiz, ebx
        invoke  con_write_asciiz, str_newline
 
; Start the search
        invoke  file.find.first, ebx, str_mask, FA_READONLY+FA_FOLDER+FA_NORMAL+FA_ARCHIVED
        test    eax, eax
        jz      .nosuchdir
 
        lea     edi, [ebp + thread_data.buffer]
  .parse_file:
        test    eax, eax        ; did we find a file?
        jz      .done
        mov     ebx, eax        ; yes, save the descripter in ebx
 
; first, convert the attributes
        test    [ebx + FileInfoA.Attributes], FA_FOLDER
        jnz     .folder
 
        test    [ebx + 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     eax, [ebx + FileInfoA.FileSizeLow]
        call    dword_to_ascii
        mov     al, ' '
        stosb
 
; then date (month/day/year)
        movzx   eax, [ebx + FileInfoA.DateModify + FileDateTime.month]
        cmp     eax, 12
        ja      @f
        mov     eax, [months - 4 + 4*eax]
        stosd
       @@:
 
        movzx   eax, [ebx + FileInfoA.DateModify + FileDateTime.day]
        call    dword_to_ascii
        mov     al, ' '
        stosb
 
        movzx   eax, [ebx + FileInfoA.DateModify + FileDateTime.year]
        call    dword_to_ascii
        mov     al, ' '
        stosb
 
; and last but not least, filename
        lea     esi, [ebx + FileInfoA.FileName]
        mov     ecx, 264
  .nameloop:
        lodsb
        test    al, al
        jz      .namedone
        stosb
        loop    .nameloop
 
; insert a cr lf
  .namedone:
        mov     ax, 0x0a0d
        stosw
 
        test    [ebp + thread_data.permissions], ABORT          ; Did we receive ABOR command from client?
;;;        jnz     .abort                                         ; TODO
 
; check next file
        invoke  file.find.next, ebx
        jmp     .parse_file
 
; close file desc
  .done:
        invoke  file.find.close, ebx                            ; ebx is descriptor of last file, eax will be -1 !!
 
; append the string with a 0
        xor     al, al
        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, [ebp + thread_data.datasocketnum]          ; socket num
        lea     edx, [ebp + thread_data.buffer]                 ; buffer ptr
        sub     esi, edx                                        ; length
        xor     edi, edi                                        ; flags
        mcall   send
 
; close the data socket..
        mov     [ebp + thread_data.mode], MODE_NOTREADY
        mcall   close, [ebp + thread_data.datasocketnum]
 
        sendFTP "226 Transfer OK"
        ret
 
  .nosuchdir:
        sendFTP "550 Directory does not exist"
        ret
 
;------------------------------------------------
; "NLST"
;
; List the filenames of the files in the current working directory.
;
;------------------------------------------------
align 4
cmdNLST:
 
        test    [ebp + thread_data.permissions], PERMISSION_EXEC
        jz      permission_denied
 
        ; TODO: same as list but simpler output format
 
        ret
 
;------------------------------------------------
; "NOOP"
;
; No operation, just keep the connection alive.
;
;------------------------------------------------
align 4
cmdNOOP:
 
        sendFTP "200 Command OK"
        ret
 
;------------------------------------------------
; "PASS"
;
; Second phase of login process, client provides password.
;
;------------------------------------------------
align 4
cmdPASS:
 
; read the password from users.ini
        lea     edi, [ebp + thread_data.buffer + 512]           ; temp pass
        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
        sub     ecx, 5
        jbe     .incorrect                                      ; no password given? but hey, we need one! fail..
 
; compare with received password
        repe    cmpsb
        cmp     byte [esi-1], 0x20                              ; printeable characters left?
        jae     .incorrect
        cmp     byte [edi-1], 0
        jne     .incorrect
 
  .ok:
        invoke  ini.get_int, path2, ebx, str_mode, 0
        mov     [ebp + thread_data.permissions], eax
 
        invoke  con_write_asciiz, str_pass_ok
        mov     [ebp + thread_data.state], STATE_ACTIVE
        sendFTP "230 You are now logged in"
        ret
 
  .2:
  .incorrect:
        invoke  con_write_asciiz, str_pass_err
        mov     [ebp + thread_data.state], STATE_CONNECTED      ; reset state
        sendFTP "530 Login incorrect"
        ret
 
  .0:
        sendFTP "503 Login with USER first"
        ret
 
  .3:
        sendFTP "230 Already logged in"
        ret
 
;------------------------------------------------
; "PASV"
;
; Initiate a passive dataconnection.
;
;------------------------------------------------
align 4
cmdPASV:
 
        cmp     [ebp + thread_data.passivesocknum], -1
        je      @f
        mcall   close, [ebp + thread_data.passivesocknum]       ; if there is still a socket open, close it
       @@:
 
; Open a new TCP socket
        mcall   socket, AF_INET4, SOCK_STREAM, 0
        cmp     eax, -1
        je      socketerror
        mov     [ebp + thread_data.passivesocknum], eax
 
; Bind it to a known local port
        mov     [ebp + thread_data.datasock.sin_family], AF_INET4
        mov     [ebp + thread_data.datasock.sin_addr], 0
 
        mov     ecx, eax                                        ; passivesocketnum
        lea     edx, [ebp + thread_data.datasock]
        mov     esi, sizeof.thread_data.datasock
 
  .next_port:                                                   ; TODO: break the endless loop
        call    nextpasvport
        pushw   [pasv_port]
        popw    [ebp + thread_data.datasock.sin_port]
 
        mcall   bind
        cmp     eax, -1
        je      .next_port
 
; And set it to listen!
        mcall   listen, , 1
        cmp     eax, -1
        je      socketerror
 
; Tell our thread we are ready to accept incoming calls
        mov     [ebp + thread_data.mode], MODE_PASSIVE_WAIT
 
; Now tell the client where to connect to in this format:
; 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.
 
; '227 ('
        lea     edi, [ebp + thread_data.buffer]
        mov     eax, '227 '
        stosd
        mov     al, '('
        stosb
; 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 + 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]
        sub     esi, ebp
        mov     ecx, [ebp + thread_data.socketnum]
        lea     edx, [ebp + thread_data.buffer]
        xor     edi, edi
        mcall   send
 
        ret
 
;------------------------------------------------
; "PWD"
;
; Print the current working directory.
;
;------------------------------------------------
align 4
cmdPWD:
 
        mov     dword [ebp + thread_data.buffer], '257 '
        mov     byte [ebp + thread_data.buffer+4], '"'
 
        lea     edi, [ebp + thread_data.buffer+5]
        lea     esi, [ebp + thread_data.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 - 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
        lea     eax, [ebp + thread_data.work_dir]
        invoke  con_write_asciiz, eax
        invoke  con_write_asciiz, str_newline
 
        ret
 
;------------------------------------------------
; "PORT"
;
; Initiate an active dataconnection.
;
;------------------------------------------------
align 4
cmdPORT:
 
; PORT a1,a2,a3,a4,p1,p2
; IP address a1.a2.a3.a4, port p1*256+p2
 
; Convert the IP
        lea     esi, [esi+5]
        mov     cl, ','
        call    ip_to_dword
; And put it in datasock
        mov     [ebp + thread_data.datasock.sin_addr], ebx
 
; Now the same with portnumber
        inc     esi
        call    ascii_to_byte
        mov     bh, al
        inc     esi
        call    ascii_to_byte
        mov     bl, al
 
; Save it in datasock too
        mov     [ebp + thread_data.datasock.sin_port], bx
 
; We will open the socket, but do not connect yet!
        mov     [ebp + thread_data.datasock.sin_family], AF_INET4
        mcall   socket, AF_INET4, SOCK_STREAM, 0
        cmp     eax, -1
        je      socketerror
 
        mov     [ebp + thread_data.datasocketnum], eax
        mov     [ebp + thread_data.mode], MODE_ACTIVE
 
        sendFTP "225 Data connection open"
        ret
 
;------------------------------------------------
; "QUIT"
;
; Close the connection with client.
;
;------------------------------------------------
align 4
cmdQUIT:
 
        sendFTP "221 Bye!"
        mcall   close, [ebp + thread_data.datasocketnum]
        mcall   close, [ebp + thread_data.socketnum]
 
        add     esp, 4          ; get rid of call return address
        jmp     thread_exit     ; now close this thread
 
 
;------------------------------------------------
; "RETR"
;
; Retrieve a file from the ftp server.
;
;------------------------------------------------
align 4
cmdRETR:
 
        test    [ebp + thread_data.permissions], PERMISSION_READ
        jz      permission_denied
 
        cmp     ecx, 1024 + 5
        jae     .cannot_open
 
        sub     ecx, 5
        jb      .cannot_open
 
        cmp     [ebp + thread_data.mode], MODE_PASSIVE_OK
        je      .start
 
; If we are in active mode, it's time to open a data socket..
        cmp     [ebp + thread_data.mode], MODE_ACTIVE
        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
        jmp     .start
 
; If we are still in passive_wait, it's time we accept an incomming call..
  .not_active:
        cmp     [ebp + thread_data.mode], MODE_PASSIVE_WAIT
        jne     socketerror
        mov     [ebp + thread_data.mode], MODE_PASSIVE_FAILED                   ; assume that we will fail
  .try_now:
        mov     ecx, [ebp + thread_data.passivesocknum]
        lea     edx, [ebp + thread_data.datasock]
        mov     esi, sizeof.thread_data.datasock
        mcall   accept
        cmp     eax, -1
        jne     .pasv_ok
        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
 
  .start:
        call    create_path
        dec     edi
        add     esi, 5
 
        lea     esi, [ebp + thread_data.buffer]                         ; FIXME
        mov     ecx, 1024
  .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, ebx
        invoke  con_write_asciiz, str_newline
 
        invoke  file.open, ebx, O_READ
        test    eax, eax
        jz      .cannot_open
 
        push    eax
        sendFTP "150 Here it comes.."
        pop     ebx
 
  .read_more:
        test    [ebp + thread_data.permissions], ABORT
        jnz     abort_transfer
 
        lea     eax, [ebp + thread_data.buffer]                 ; FIXME: use another buffer!! if we receive something on control connection now, we screw up!
        invoke  file.read, ebx, eax, BUFFERSIZE
        cmp     eax, -1
        je      .cannot_open                                    ; FIXME: this is not the correct error
 
        push    eax ebx
        mov     esi, eax
        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
;        jne     not_all_byes_sent                               ; TODO
 
        cmp     ecx, BUFFERSIZE
        je      .read_more
 
        invoke  file.close, ebx
 
        mov     [ebp + thread_data.mode], MODE_NOTREADY
        mcall   close, [ebp + thread_data.datasocketnum]
 
        sendFTP "226 Transfer OK, closing connection"
        ret
 
  .cannot_open:
        invoke  con_set_flags, 0x0c
        invoke  con_write_asciiz, str_notfound
        invoke  con_set_flags, 0x07
 
        sendFTP "550 No such file"
        ret
 
 
 
;------------------------------------------------
; "STOR"
;
; Store a file on the server.
;
;------------------------------------------------
align 4
cmdSTOR:
 
        test    [ebp + thread_data.permissions], PERMISSION_WRITE
        jz      permission_denied
 
 
;;;;
        test    [ebp + thread_data.permissions], ABORT
        jnz     abort_transfer
 
;;;;
 
        ret
 
;------------------------------------------------
; "SYST"
;
; Send information about the system.
;
;------------------------------------------------
align 4
cmdSYST:
 
        sendFTP "215 UNIX type: L8"
        ret
 
;------------------------------------------------
; "TYPE"
;
; Choose the file transfer type.
;
;------------------------------------------------
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     [ebp + thread_data.type], TYPE_ASCII
        jmp     .subtype
 
  .ebdic:
        mov     [ebp + thread_data.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      [ebp + thread_data.type], TYPE_NP
        jmp     .ok
 
  .telnet:
        or      [ebp + thread_data.type], TYPE_TELNET
        jmp     .ok
 
  .asacc:
        or      [ebp + thread_data.type], TYPE_ASA
        jmp     .ok
 
  .image:
        mov     [ebp + thread_data.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                         ; 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:
        sendFTP "200 Command ok"
        ret
 
;------------------------------------------------
; "USER"
;
; Login to the server, step one of two.                         ;;; TODO: prevent buffer overflow!
;
;------------------------------------------------
align 4
cmdUSER:
 
        lea     esi, [esi + 5]
        lea     edi, [ebp + thread_data.fpath]                  ; temp buffer for username
  .loop:
        lodsb
        stosb
        cmp     al, 0x20
        jae     .loop
        mov     byte [edi-1], 0
 
        lea     esi, [ebp + thread_data.fpath]
        lea     eax, [ebp + thread_data.home_dir]
        invoke  ini.get_str, path2, esi, str_home, eax, 1024, str_infinity
        cmp     eax, -1
        je      .login_fail
        cmp     dword [esi], -1
        je      .login_fail
 
        mov     word [ebp + thread_data.work_dir], "/"          ; "/", 0
 
        invoke  con_write_asciiz, str_logged_in
        mov     [ebp + thread_data.state], STATE_LOGIN
  .sendstr:
        sendFTP "331 Please specify the password"
        ret
 
  .login_fail:
        invoke  con_write_asciiz, str_pass_err
        mov     [ebp + thread_data.state], STATE_LOGIN_FAIL
        jmp     .sendstr
 
align 4
  .2:
        sendFTP "530 Can't change to another user"
        ret

Rev 2624	Rev 2635
1	struct thread_data	1	struct thread_data
2	rb 1024	2	rb 1024
3	stack rb 0	3	stack rb 0
4		4
5	home_dir rb 1024 ; home directory in wich the user is locked, asciiz	5	home_dir rb 1024 ; home directory in wich the user is locked, asciiz
6	work_dir rb 1024 ; working directory, must at all times begin and end with a '/', asciiz	6	work_dir rb 1024 ; working directory, must at all times begin and end with a '/', asciiz
7	fpath rb 1024*3 ; file path, combination of home_dir, work_dir and filename	7	fpath rb 1024*3 ; file path, combination of home_dir, work_dir and filename
8	; Will also be used to temporarily store username	8	; Will also be used to temporarily store username
9		9
10	type db ? ; ASCII/EBDIC/IMAGE/..	10	type db ? ; ASCII/EBDIC/IMAGE/..
11	mode db ? ; active/passive	11	mode db ? ; active/passive
12	socketnum dd ? ; Commands socket	12	socketnum dd ? ; Commands socket
13	state dd ? ; disconnected/logging in/logged in/..	13	state dd ? ; disconnected/logging in/logged in/..
14	passivesocknum dd ? ; when in passive mode, this is the listening socket	14	passivesocknum dd ? ; when in passive mode, this is the listening socket
15	datasocketnum dd ? ; socket used for data transfers	15	datasocketnum dd ? ; socket used for data transfers
16	permissions dd ? ; read/write/execute/....	16	permissions dd ? ; read/write/execute/....
17	buffer_ptr dd ?	17	buffer_ptr dd ?
18	pid dd ? ; Process id of the current thread	18	pid dd ? ; Process id of the current thread
19		19
20	datasock sockaddr_in	20	datasock sockaddr_in
21		21
22	buffer rb BUFFERSIZE	22	buffer rb BUFFERSIZE
23	ends	23	ends
24		24
25		25	;------------------------------------------------
26	macro sendFTP str {	-
27	local .string, .length, .label	-
28	xor edi, edi	-
29	mcall send, [ebp + thread_data.socketnum], .string, .length	-
30	jmp @f	-
31	.string db str, 13, 10	-
32	.length = $ - .string	-
33	@@:	-
34		-
35	}	-
36		-
37	;------------------------------------------------	-
38	; parse_cmd	26	; parse_cmd
39	;	27	;
40	; Internal function wich uses the 'commands'	28	; Internal function wich uses the 'commands'
41	; table to call an appropriate cmd_xx function.	29	; table to call an appropriate cmd_xx function.
42	;	30	;
43	; input: esi = ptr to ascii commands	31	; input: esi = ptr to ascii commands
44	; ecx = number of bytes input	32	; ecx = number of bytes input
45	; ebp = pointer to thread_data structure	33	; ebp = pointer to thread_data structure
46	;	34	;
47	; output: none	35	; output: none
48	;	36	;
49	;------------------------------------------------	37	;------------------------------------------------
50	align 4	38	align 4
51	parse_cmd: ; esi must point to command	39	parse_cmd: ; esi must point to command
52		40
53	cmp byte [esi], 0x20 ; skip all leading characters	41	cmp byte [esi], 0x20 ; skip all leading characters
54	ja .ok	42	ja .ok
55	inc esi	43	inc esi
56	dec ecx	44	dec ecx
57	cmp ecx, 3	45	cmp ecx, 3
58	jb .error	46	jb .error
59	jmp parse_cmd	47	jmp parse_cmd
60	.ok:	48	.ok:
61	cmp byte [esi+3], 0x20	49	cmp byte [esi+3], 0x20
62	ja @f	50	ja @f
63	mov byte [esi+3], 0	51	mov byte [esi+3], 0
64	@@:	52	@@:
65		53
66	mov eax, [esi]	54	mov eax, [esi]
67	and eax, not 0x20202020 ; convert to upper case	55	and eax, not 0x20202020 ; convert to upper case
68	mov edi, commands ; list of commands to scan	56	mov edi, commands ; list of commands to scan
69	.scanloop:	57	.scanloop:
70	cmp eax, [edi]	58	cmp eax, [edi]
71	je .got_it	59	je .got_it
72		60
73	add edi, 5*4	61	add edi, 5*4
74	cmp byte [edi], 0	62	cmp byte [edi], 0
75	jne .scanloop	63	jne .scanloop
76		64
77	.error:	65	.error:
78	cmp [ebp + thread_data.state], STATE_ACTIVE	66	cmp [ebp + thread_data.state], STATE_ACTIVE
79	jb login_first	67	jb login_first
80	sendFTP "500 Unsupported command"	68	sendFTP "500 Unsupported command"
81	ret	69	ret
82		70
83	.got_it:	71	.got_it:
84	mov eax, [ebp + thread_data.state]	72	mov eax, [ebp + thread_data.state]
85	jmp dword [edi + 4 + eax]	73	jmp dword [edi + 4 + eax]
86		74
87		75
88	align 4	76	align 4
89	commands: ; all commands must be in uppercase	77	iglobal
90		78	commands: ; all commands must be in uppercase
-		79
91	dd 'ABOR', login_first, login_first, login_first, cmdABOR	80	dd 'ABOR', login_first, login_first, login_first, cmdABOR
92	; dd 'ACCT', login_first, login_first, login_first, cmd_ACCT	81	; dd 'ACCT', login_first, login_first, login_first, cmd_ACCT
93	; dd 'APPE', login_first, login_first, login_first, cmd_APPE	82	; dd 'APPE', login_first, login_first, login_first, cmd_APPE
94	dd 'CDUP', login_first, login_first, login_first, cmdCDUP	83	dd 'CDUP', login_first, login_first, login_first, cmdCDUP
95	dd 'CWD', login_first, login_first, login_first, cmdCWD	84	dd 'CWD', login_first, login_first, login_first, cmdCWD
96	dd 'DELE', login_first, login_first, login_first, cmdDELE	85	dd 'DELE', login_first, login_first, login_first, cmdDELE
97	; dd 'HELP', login_first, login_first, login_first, cmd_HELP	86	; dd 'HELP', login_first, login_first, login_first, cmd_HELP
98	dd 'LIST', login_first, login_first, login_first, cmdLIST	87	dd 'LIST', login_first, login_first, login_first, cmdLIST
99	; dd 'MDTM', login_first, login_first, login_first, cmd_MDTM	88	; dd 'MDTM', login_first, login_first, login_first, cmd_MDTM
100	; dd 'MKD', login_first, login_first, login_first, cmd_MKD	89	; dd 'MKD', login_first, login_first, login_first, cmd_MKD
101	; dd 'MODE', login_first, login_first, login_first, cmd_MODE	90	; dd 'MODE', login_first, login_first, login_first, cmd_MODE
102	dd 'NLST', login_first, login_first, login_first, cmdNLST	91	dd 'NLST', login_first, login_first, login_first, cmdNLST
103	dd 'NOOP', login_first, login_first, login_first, cmdNOOP	92	dd 'NOOP', login_first, login_first, login_first, cmdNOOP
104	dd 'PASS', cmdPASS.0, cmdPASS , cmdPASS.2, cmdPASS.3	93	dd 'PASS', cmdPASS.0, cmdPASS , cmdPASS.2, cmdPASS.3
105	dd 'PASV', login_first, login_first, login_first, cmdPASV	94	dd 'PASV', login_first, login_first, login_first, cmdPASV
106	dd 'PORT', login_first, login_first, login_first, cmdPORT	95	dd 'PORT', login_first, login_first, login_first, cmdPORT
107	dd 'PWD', login_first, login_first, login_first, cmdPWD	96	dd 'PWD', login_first, login_first, login_first, cmdPWD
108	dd 'QUIT', cmdQUIT, cmdQUIT, cmdQUIT, cmdQUIT	97	dd 'QUIT', cmdQUIT, cmdQUIT, cmdQUIT, cmdQUIT
109	; dd 'REIN', login_first, login_first, login_first, cmd_REIN	98	; dd 'REIN', login_first, login_first, login_first, cmd_REIN
110	; dd 'REST', login_first, login_first, login_first, cmd_REST	99	; dd 'REST', login_first, login_first, login_first, cmd_REST
111	dd 'RETR', login_first, login_first, login_first, cmdRETR	100	dd 'RETR', login_first, login_first, login_first, cmdRETR
112	; dd 'RMD', login_first, login_first, login_first, cmd_RMD	101	; dd 'RMD', login_first, login_first, login_first, cmd_RMD
113	; dd 'RNFR', login_first, login_first, login_first, cmd_RNFR	102	; dd 'RNFR', login_first, login_first, login_first, cmd_RNFR
114	; dd 'RNTO', login_first, login_first, login_first, cmd_RNTO	103	; dd 'RNTO', login_first, login_first, login_first, cmd_RNTO
115	; dd 'SITE', login_first, login_first, login_first, cmd_SITE	104	; dd 'SITE', login_first, login_first, login_first, cmd_SITE
116	; dd 'SIZE', login_first, login_first, login_first, cmd_SIZE	105	; dd 'SIZE', login_first, login_first, login_first, cmd_SIZE
117	; dd 'STAT', login_first, login_first, login_first, cmd_STAT	106	; dd 'STAT', login_first, login_first, login_first, cmd_STAT
118	dd 'STOR', login_first, login_first, login_first, cmdSTOR	107	dd 'STOR', login_first, login_first, login_first, cmdSTOR
119	; dd 'STOU', login_first, login_first, login_first, cmd_STOU	108	; dd 'STOU', login_first, login_first, login_first, cmd_STOU
120	; dd 'STRU', login_first, login_first, login_first, cmd_STRU	109	; dd 'STRU', login_first, login_first, login_first, cmd_STRU
121	dd 'SYST', login_first, login_first, login_first, cmdSYST	110	dd 'SYST', login_first, login_first, login_first, cmdSYST
122	dd 'TYPE', login_first, login_first, login_first, cmdTYPE	111	dd 'TYPE', login_first, login_first, login_first, cmdTYPE
123	dd 'USER', cmdUSER, cmdUSER, cmdUSER, cmdUSER.2	112	dd 'USER', cmdUSER, cmdUSER, cmdUSER, cmdUSER.2
124	db 0 ; end marker	113	db 0 ; end marker
125		114	endg
126	align 4	115
-		116	align 4
127	login_first:	117	login_first:
128	sendFTP "530 Please login with USER and PASS"	118	sendFTP "530 Please login with USER and PASS"
129	ret	119	ret
130		120
131	align 4	121	align 4
132	permission_denied:	122	permission_denied:
133	sendFTP "550 Permission denied"	123	sendFTP "550 Permission denied"
134	ret	124	ret
135		125
136	align 4	126	align 4
137	socketerror:	127	socketerror:
138	invoke con_set_flags, 0x0c	128	invoke con_set_flags, 0x0c
139	invoke con_write_asciiz, str_sockerr	129	invoke con_write_asciiz, str_sockerr
140	invoke con_set_flags, 0x07	130	invoke con_set_flags, 0x07
141		131
142	sendFTP "425 Can't open data connection"	132	sendFTP "425 Can't open data connection"
143	ret	133	ret
144		134
145	align 4	135	align 4
146	abort_transfer:	136	abort_transfer:
147	and [ebp + thread_data.permissions], not ABORT	137	and [ebp + thread_data.permissions], not ABORT
148	mov [ebp + thread_data.mode], MODE_NOTREADY	138	mov [ebp + thread_data.mode], MODE_NOTREADY
149	invoke file.close, ebx	139	invoke file.close, ebx
150	mcall close, [ebp + thread_data.datasocketnum]	140	mcall close, [ebp + thread_data.datasocketnum]
151		141
152	sendFTP "530 Transfer aborted"	142	sendFTP "530 Transfer aborted"
153	ret	143	ret
154		144
155	align 4	145	align 4
156	ip_to_dword: ; esi = ptr to str, cl = separator ('.', ',')	146	ip_to_dword: ; esi = ptr to str, cl = separator ('.', ',')
157		147
158	call ascii_to_byte	148	call ascii_to_byte
159	mov bl, al	149	mov bl, al
160	cmp byte [esi], cl	150	cmp byte [esi], cl
161	jne .err	151	jne .err
162	inc esi	152	inc esi
163		153
164	call ascii_to_byte	154	call ascii_to_byte
165	mov bh, al	155	mov bh, al
166	cmp byte [esi], cl	156	cmp byte [esi], cl
167	jne .err	157	jne .err
168	inc esi	158	inc esi
169		159
170	shl ebx, 16	160	shl ebx, 16
171		161
172	call ascii_to_byte	162	call ascii_to_byte
173	mov bl, al	163	mov bl, al
174	cmp byte [esi], cl	164	cmp byte [esi], cl
175	jne .err	165	jne .err
176	inc esi	166	inc esi
177		167
178	call ascii_to_byte	168	call ascii_to_byte
179	mov bh, al	169	mov bh, al
180		170
181	ror ebx, 16	171	ror ebx, 16
182	ret	172	ret
183		173
184	.err:	174	.err:
185	xor ebx, ebx	175	xor ebx, ebx
186	ret	176	ret
187		177
188	align 4 ; esi = ptr to str, output in eax	178	align 4 ; esi = ptr to str, output in eax
189	ascii_to_byte:	179	ascii_to_byte:
190		180
191	xor eax, eax	181	xor eax, eax
192	push ebx	182	push ebx
193		183
194	.loop:	184	.loop:
195	movzx ebx, byte[esi]	185	movzx ebx, byte[esi]
196	sub bl, '0'	186	sub bl, '0'
197	jb .done	187	jb .done
198	cmp bl, 9	188	cmp bl, 9
199	ja .done	189	ja .done
200	lea eax, [eax*4 + eax] ;	190	lea eax, [eax*4 + eax] ;
201	shl eax, 1 ; eax = eax * 10	191	shl eax, 1 ; eax = eax * 10
202	add eax, ebx	192	add eax, ebx
203	inc esi	193	inc esi
204		194
205	jmp .loop	195	jmp .loop
206		196
207	.done:	197	.done:
208	pop ebx	198	pop ebx
209	ret	199	ret
210		200
211	align 4	201	align 4
212	dword_to_ascii: ; edi = ptr where to write, eax is number	202	dword_to_ascii: ; edi = ptr where to write, eax is number
213		203
214	push edx ebx ecx	204	push edx ebx ecx
215	mov ebx, 10	205	mov ebx, 10
216	xor ecx, ecx	206	xor ecx, ecx
217		207
218	@@:	208	@@:
219	xor edx, edx	209	xor edx, edx
220	div ebx	210	div ebx
221	add edx, '0'	211	add edx, '0'
222	pushw dx	212	pushw dx
223	inc ecx	213	inc ecx
224	test eax, eax	214	test eax, eax
225	jnz @r	215	jnz @r
226		216
227	@@:	217	@@:
228	popw ax	218	popw ax
229	stosb	219	stosb
230	dec ecx	220	dec ecx
231	jnz @r	221	jnz @r
232		222
233	pop ecx ebx edx	223	pop ecx ebx edx
234	ret	224	ret
235		225
236	align 4	226	align 4
237	create_path: ; combine home_dir and work_dir strings into fpath	227	create_path: ; combine home_dir and work_dir strings into fpath
238		228
239	lea edi, [ebp + thread_data.fpath]	229	lea edi, [ebp + thread_data.fpath]
240	lea esi, [ebp + thread_data.home_dir]	230	lea esi, [ebp + thread_data.home_dir]
241	mov ecx, 1024	231	mov ecx, 1024
242	.loop1:	232	.loop1:
243	lodsb	233	lodsb
244	cmp al, 0x20	234	cmp al, 0x20
245	jb .next	235	jb .next
246	stosb	236	stosb
247	loop .loop1	237	loop .loop1
248	.next:	238	.next:
249		239
250	cmp byte[edi-1], '/'	240	cmp byte[edi-1], '/'
251	jne @f	241	jne @f
252	dec edi	242	dec edi
253	@@:	243	@@:
254		244
255	lea esi, [ebp + thread_data.work_dir]	245	lea esi, [ebp + thread_data.work_dir]
256	mov ecx, 1024	246	mov ecx, 1024
257	.loop2:	247	.loop2:
258	lodsb	248	lodsb
259	cmp al, 0x20	249	cmp al, 0x20
260	jb .done	250	jb .done
261	stosb	251	stosb
262	loop .loop2	252	loop .loop2
263		253
264	.done:	254	.done:
265	xor al, al	255	xor al, al
266	stosb	256	stosb
267	ret	257	ret
268		258
269		259
270	align 4	260	align 4
271	nextpasvport:	261	nextpasvport:
272		262
273	inc [pasv_port]	263	inc [pasv_port]
274		264
275	mov ax, [pasv_port]	265	mov ax, [pasv_port]
276	cmp ax, [pasv_start]	266	cmp ax, [pasv_start]
277	jb .restart	267	jb .restart
278	cmp ax, [pasv_end]	268	cmp ax, [pasv_end]
279	ja .restart	269	ja .restart
280		270
281	ret	271	ret
282		272
283	.restart:	273	.restart:
284	pushw [pasv_start]	274	pushw [pasv_start]
285	popw [pasv_port]	275	popw [pasv_port]
286		276
287	ret	277	ret
288		278
289		279
290	;------------------------------------------------	280	;------------------------------------------------
291	; "ABOR"	281	; "ABOR"
292	;	282	;
293	; This command aborts the current filetransfer.	283	; This command aborts the current filetransfer.
294	;	284	;
295	;------------------------------------------------	285	;------------------------------------------------
296	align 4	286	align 4
297	cmdABOR:	287	cmdABOR:
298		288
299	or [ebp + thread_data.permissions], ABORT	289	or [ebp + thread_data.permissions], ABORT
300	sendFTP "250 Command succesul"	290	sendFTP "250 Command succesul"
301	ret	291	ret
302		292
303	;------------------------------------------------	293	;------------------------------------------------
304	; "CDUP"	294	; "CDUP"
305	;	295	;
306	; Change the directory to move up one level.	296	; Change the directory to move up one level.
307	;	297	;
308	;------------------------------------------------	298	;------------------------------------------------
309	align 4	299	align 4
310	cmdCDUP:	300	cmdCDUP:
311		301
312	test [ebp + thread_data.permissions], PERMISSION_CD	302	test [ebp + thread_data.permissions], PERMISSION_CD
313	jz permission_denied	303	jz permission_denied
314		304
315	cmp byte [ebp + thread_data.work_dir+1], 0 ; are we in "/" ?	305	cmp byte [ebp + thread_data.work_dir+1], 0 ; are we in "/" ?
316	je .done ; if so, we cant go up..	306	je .done ; if so, we cant go up..
317		307
318	; find the end of asciiz string work_dir	308	; find the end of asciiz string work_dir
319	mov ecx, 1024	309	mov ecx, 1024
320	xor al, al	310	xor al, al
321	lea edi, [ebp + thread_data.work_dir]	311	lea edi, [ebp + thread_data.work_dir]
322	repne scasb	312	repne scasb
323	; return 2 characters (right before last /)	313	; return 2 characters (right before last /)
324	sub edi, 3	314	sub edi, 3
325	; and now search backwards, for a '/'	315	; and now search backwards, for a '/'
326	mov al,'/'	316	mov al,'/'
327	neg ecx	317	neg ecx
328	add ecx, 1024	318	add ecx, 1024
329	std	319	std
330	repne scasb	320	repne scasb
331	cld	321	cld
332	; terminate the string here	322	; terminate the string here
333	mov byte[edi+2], 0	323	mov byte[edi+2], 0
334		324
335	.done:	325	.done:
336	; Print the new working dir on the console	326	; Print the new working dir on the console
337	lea eax, [ebp + thread_data.work_dir]	327	lea eax, [ebp + thread_data.work_dir]
338	invoke con_write_asciiz, eax	328	invoke con_write_asciiz, eax
339	invoke con_write_asciiz, str_newline	329	invoke con_write_asciiz, str_newline
340		330
341	sendFTP "250 Command succesul"	331	sendFTP "250 Command succesul"
342	ret	332	ret
343		333
344	;------------------------------------------------	334	;------------------------------------------------
345	; "CWD"	335	; "CWD"
346	;	336	;
347	; Change Working Directory.	337	; Change Working Directory.
348	;	338	;
349	;------------------------------------------------	339	;------------------------------------------------
350	align 4	340	align 4
351	cmdCWD:	341	cmdCWD:
352		342
353	test [ebp + thread_data.permissions], PERMISSION_CD	343	test [ebp + thread_data.permissions], PERMISSION_CD
354	jz permission_denied	344	jz permission_denied
355		345
356	; do we have enough parameters?	346	; do we have enough parameters?
357	sub ecx, 4	347	sub ecx, 4
358	jbe .err	348	jbe .err
359		349
360	; get ready to copy the path	350	; get ready to copy the path
361	add esi, 4	351	add esi, 4
362	mov ecx, 1024	352	mov ecx, 1024
363	lea edi, [ebp + thread_data.work_dir]	353	lea edi, [ebp + thread_data.work_dir]
364		354
365	; if received dir starts with '/', we will simply copy it	355	; if received dir starts with '/', we will simply copy it
366	; If not, we will append the current path with the received path.	356	; If not, we will append the current path with the received path.
367	cmp byte [esi], '/'	357	cmp byte [esi], '/'
368	je .copyloop	358	je .copyloop
369		359
370	; Find the end of work_dir string.	360	; Find the end of work_dir string.
371	xor al, al	361	xor al, al
372	.find_zero:	362	.find_zero:
373	repne scasb	363	repne scasb
374	dec edi	364	dec edi
375		365
376	; and now append work_dir with received string	366	; and now append work_dir with received string
377	mov ecx, 1024	367	mov ecx, 1024
378		368
379	; scan for end byte, or '.'	369	; scan for end byte, or '.'
380	.copyloop:	370	.copyloop:
381	lodsb	371	lodsb
382	cmp al, 0x20	372	cmp al, 0x20
383	jb .done	373	jb .done
384	;;; cmp al, '.' ; '..' means we must go up one dir TODO	374	;;; cmp al, '.' ; '..' means we must go up one dir TODO
385	;;; je .up	375	;;; je .up
386	stosb	376	stosb
387	loop .copyloop	377	loop .copyloop
388		378
389	; now, now make sure it ends with '/', 0	379	; now, now make sure it ends with '/', 0
390	.done:	380	.done:
391	cmp byte [edi-1], '/'	381	cmp byte [edi-1], '/'
392	je @f	382	je @f
393	mov byte [edi], '/'	383	mov byte [edi], '/'
394	inc edi	384	inc edi
395	@@:	385	@@:
396	mov byte [edi], 0	386	mov byte [edi], 0
397		387
398	; Print the new working dir on the console	388	; Print the new working dir on the console
399	lea eax, [ebp + thread_data.work_dir]	389	lea eax, [ebp + thread_data.work_dir]
400	invoke con_write_asciiz, eax	390	invoke con_write_asciiz, eax
401	invoke con_write_asciiz, str_newline	391	invoke con_write_asciiz, str_newline
402		392
403	sendFTP "250 Command succesful"	393	sendFTP "250 Command succesful"
404	ret	394	ret
405		395
406	.err:	396	.err:
407	sendFTP "550 Directory does not exist"	397	sendFTP "550 Directory does not exist"
408	ret	398	ret
409		399
410	;------------------------------------------------	400	;------------------------------------------------
411	; "DELE"	401	; "DELE"
412	;	402	;
413	; Delete a file from the server.	403	; Delete a file from the server.
414	;	404	;
415	;------------------------------------------------	405	;------------------------------------------------
416	align 4	406	align 4
417	cmdDELE:	407	cmdDELE:
418		408
419	test [ebp + thread_data.permissions], PERMISSION_DELETE	409	test [ebp + thread_data.permissions], PERMISSION_DELETE
420	jz permission_denied	410	jz permission_denied
421		411
422	ret	412	ret
423		413
424	;------------------------------------------------	414	;------------------------------------------------
425	; "LIST"	415	; "LIST"
426	;	416	;
427	; List the files in the current working directory.	417	; List the files in the current working directory.
428	;	418	;
429	;------------------------------------------------	419	;------------------------------------------------
430	align 4	420	align 4
431	cmdLIST:	421	cmdLIST:
432		422
433	test [ebp + thread_data.permissions], PERMISSION_EXEC	423	test [ebp + thread_data.permissions], PERMISSION_EXEC
434	jz permission_denied	424	jz permission_denied
435		425
436	cmp [ebp + thread_data.mode], MODE_PASSIVE_OK	426	cmp [ebp + thread_data.mode], MODE_PASSIVE_OK
437	je .start	427	je .start
438		428
439	; If we are in active mode, it's time to open a data socket..	429	; If we are in active mode, it's time to open a data socket..
440	cmp [ebp + thread_data.mode], MODE_ACTIVE	430	cmp [ebp + thread_data.mode], MODE_ACTIVE
441	jne .not_active	431	jne .not_active
442	mov ecx, [ebp + thread_data.datasocketnum]	432	mov ecx, [ebp + thread_data.datasocketnum]
443	lea edx, [ebp + thread_data.datasock]	433	lea edx, [ebp + thread_data.datasock]
444	mov esi, sizeof.thread_data.datasock	434	mov esi, sizeof.thread_data.datasock
445	mcall connect	435	mcall connect
446	cmp eax, -1	436	cmp eax, -1
447	je socketerror	437	je socketerror
448	jmp .start	438	jmp .start
449		439
450	; If we are still in passive_wait, it's time we accept an incomming call..	440	; If we are still in passive_wait, it's time we accept an incomming call..
451	.not_active:	441	.not_active:
452	cmp [ebp + thread_data.mode], MODE_PASSIVE_WAIT	442	cmp [ebp + thread_data.mode], MODE_PASSIVE_WAIT
453	jne socketerror	443	jne socketerror
454	.try_now:	444	.try_now:
455	mov ecx, [ebp + thread_data.passivesocknum]	445	mov ecx, [ebp + thread_data.passivesocknum]
456	lea edx, [ebp + thread_data.datasock]	446	lea edx, [ebp + thread_data.datasock]
457	mov esi, sizeof.thread_data.datasock	447	mov esi, sizeof.thread_data.datasock
458	mcall accept	448	mcall accept
459	cmp eax, -1	449	cmp eax, -1
460	jne .pasv_ok	450	jne .pasv_ok
461	mov [ebp + thread_data.mode], MODE_PASSIVE_FAILED ; assume that we will fail	451	mov [ebp + thread_data.mode], MODE_PASSIVE_FAILED ; assume that we will fail
462	mcall 23, 200	452	mcall 23, 200
463	mcall accept	453	mcall accept
464	cmp eax, -1	454	cmp eax, -1
465	je socketerror	455	je socketerror
466	.pasv_ok:	456	.pasv_ok:
467	mov [ebp + thread_data.datasocketnum], eax	457	mov [ebp + thread_data.datasocketnum], eax
468	mov [ebp + thread_data.mode], MODE_PASSIVE_OK	458	mov [ebp + thread_data.mode], MODE_PASSIVE_OK
469	mcall close ; [ebp + thread_data.passivesocknum]	459	mcall close ; [ebp + thread_data.passivesocknum]
470	mov [ebp + thread_data.passivesocknum], -1	460	mov [ebp + thread_data.passivesocknum], -1
471	invoke con_write_asciiz, str_datasock	461	invoke con_write_asciiz, str_datasock
472		462
473	.start:	463	.start:
474	; Create fpath from home_dir and work_dir	464	; Create fpath from home_dir and work_dir
475	call create_path	465	call create_path
476		466
477	lea ebx, [ebp + thread_data.fpath]	467	lea ebx, [ebp + thread_data.fpath]
478	invoke con_write_asciiz, ebx	468	invoke con_write_asciiz, ebx
479	invoke con_write_asciiz, str_newline	469	invoke con_write_asciiz, str_newline
480		470
481	; Start the search	471	; Start the search
482	invoke file.find.first, ebx, str_mask, FA_READONLY+FA_FOLDER+FA_NORMAL+FA_ARCHIVED	472	invoke file.find.first, ebx, str_mask, FA_READONLY+FA_FOLDER+FA_NORMAL+FA_ARCHIVED
483	test eax, eax	473	test eax, eax
484	jz .nosuchdir	474	jz .nosuchdir
485		475
486	lea edi, [ebp + thread_data.buffer]	476	lea edi, [ebp + thread_data.buffer]
487	.parse_file:	477	.parse_file:
488	test eax, eax ; did we find a file?	478	test eax, eax ; did we find a file?
489	jz .done	479	jz .done
490	mov ebx, eax ; yes, save the descripter in ebx	480	mov ebx, eax ; yes, save the descripter in ebx
491		481
492	; first, convert the attributes	482	; first, convert the attributes
493	test [ebx + FileInfoA.Attributes], FA_FOLDER	483	test [ebx + FileInfoA.Attributes], FA_FOLDER
494	jnz .folder	484	jnz .folder
495		485
496	test [ebx + FileInfoA.Attributes], FA_READONLY	486	test [ebx + FileInfoA.Attributes], FA_READONLY
497	jnz .readonly	487	jnz .readonly
498		488

Subversion Repositories Kolibri OS

(root)/kernel/branches/net/applications/ftpd/commands.inc – Rev 2624 → 2635