kolibrios/kernel/branches/net/applications/ftpd/ftpd.asm

;
; Kolibrios FTP Daemon
;
; hidnplayr@gmail.com
;
; GPLv2
;

BUFFERSIZE              = 4096

STATE_DISCONNECTED      = 0
STATE_CONNECTED         = 1
STATE_LOGIN             = 2
STATE_ACTIVE            = 3

TYPE_UNDEF              = 0

TYPE_ASCII              = 00000100b
TYPE_EBDIC              = 00001000b
; subtypes for ascii & ebdic (np = default)
TYPE_NP                 = 00000001b     ; non printable
TYPE_TELNET             = 00000010b
TYPE_ASA                = 00000011b

TYPE_IMAGE              = 01000000b     ; binary data
TYPE_LOCAL              = 10000000b     ; bits per byte must be specified
                                        ; lower 4 bits will hold this value

MODE_NOTREADY           = 0
MODE_ACTIVE             = 1
MODE_PASSIVE_WAIT       = 2
MODE_PASSIVE_OK         = 3

use32
        db      'MENUET01'      ; signature
        dd      1               ; header version
        dd      start           ; entry point
        dd      i_end           ; initialized size
        dd      mem+0x1000      ; required memory
        dd      mem+0x1000      ; stack pointer
        dd      0               ; parameters
        dd      path            ; path

include '../macros.inc'
purge mov,add,sub
include '../proc32.inc'
include '../dll.inc'
include '../struct.inc'
include '../libio.inc'

include '../network.inc'
include 'commands.inc'

align 4
start:
; load libraries
        stdcall dll.Load, @IMPORT
        test    eax, eax
        jnz     exit

; find path to main settings file
        mov     edi, path      ; Calculate the length of zero-terminated string
        xor     al , al
        mov     ecx, 1024
        repne   scasb
        dec     edi
        mov     esi, filename
        movsd
        movsb

; initialize console
        push    1
        call    [con_start]

        push    title
        push    -1
        push    -1
        push    -1
        push    -1
        call    [con_init]

        mcall   40, 1 shl 7     ; we only want network events

        invoke  ini.get_int, path, str_ftpd, str_port, 21
        mov     [sockaddr1.port], ax

        push    eax
        push    str1
        call    [con_printf]

        mcall   socket, AF_INET4, SOCK_STREAM, 0
        cmp     eax, -1
        je      sock_err

        mov     [socketnum], eax

        push    str2
        call    [con_write_asciiz]

;;        mcall   setsockopt, [socketnum], SOL_SOCKET, SO_REUSEADDR, &yes,
;;        cmp     eax, -1
;;        je      opt_err

        mcall   bind, [socketnum], sockaddr1, sockaddr1.length
        cmp     eax, -1
        je      bind_err

        push    str2
        call    [con_write_asciiz]

        invoke  ini.get_int, path, str_ftpd, str_conn, 1        ; Backlog (max connections)
        mov     edx, eax

        push    str2
        call    [con_write_asciiz]

        mcall   listen, [socketnum]
        cmp     eax, -1
        je      listen_err

        push    str2b
        call    [con_write_asciiz]

        mcall   10

        mcall   accept, [socketnum], sockaddr1, sockaddr1.length
        cmp     eax, -1
        je      acpt_err

        mov     [socketnum2], eax

        mcall   send, [socketnum2], str220, str220.length, 0    ; send welcome string

  .loop:
        mcall   10

        cmp     [mode], MODE_PASSIVE_WAIT
        jne     @f
        mcall   accept, [passivesocknum], datasock, datasock.length
        cmp     eax, -1
        je      @f
        mov     [datasocketnum], eax
        mov     [mode], MODE_PASSIVE_OK

        push    str_datasock
        call    [con_write_asciiz]
       @@:

        mcall   recv, [socketnum2], buffer, buffer.length
        cmp     eax, -1
        je      .loop
        or      eax, eax
        jz      .loop
        push    eax

        mov     byte[buffer+eax], 0

        pushd   0x0a
        call    [con_set_flags]

        push    buffer
        call    [con_write_asciiz]

        pushd   0x07
        call    [con_set_flags]

        pop     ecx
        mov     esi, buffer
        call    parse_cmd

        jmp     .loop

acpt_err:

        pushd   0x0c
        call    [con_set_flags]

        push    str8
        call    [con_write_asciiz]
        jmp     done

listen_err:

        pushd   0x0c
        call    [con_set_flags]

        push    str3
        call    [con_write_asciiz]
        jmp     done

bind_err:

        pushd   0x0c
        call    [con_set_flags]

        push    str4
        call    [con_write_asciiz]
        jmp     done

sock_err:

        pushd   0x0c
        call    [con_set_flags]

        push    str6
        call    [con_write_asciiz]
        jmp     done

done:
        call    [con_getch2]
        push    1
        call    [con_exit]
exit:
        mcall   -1



; data
title   db      'KolibriOS FTP daemon 0.1', 0
str1    db      'Starting FTP daemon on port %u', 0
str2    db      '.', 0
str2b   db      ' OK!',10,10,0
str3    db      'Listen error',10,10,0
str4    db      'Bind error',10,10,0
str5    db      'Setsockopt error.',10,10,0
str6    db      'Could not open socket',10,10,0
str7    db      'Got data!',10,10,0
str8    db      'Error accepting connection',10,10,0

str_logged_in   db 'Login ok',10,10,0
str_pass_ok     db 'Password ok - Logged in',10,10,0
str_pwd         db 'Current directory is "%s"\n',0
str_err1        db 'ERROR: cannot connect to remote socket',10,10,0
str_err2        db 'ERROR: cannot open directory',10,10,0
str_datasock    db 'Passive data socket connected!',10,10,0


str_mask        db '*', 0


months:
        dd     'Jan ','Feb ','Mar ','Apr ','May ','Jun '
        dd     'Jul ','Aug ','Sep ','Oct ','Nov ','Dec '

filename db '.ini', 0
str_port db 'port', 0
str_ftpd db 'ftpd', 0
str_conn db 'conn', 0

sockaddr1:
        dw AF_INET4
.port   dw 21
.ip     dd 0
        rb 10
.length = $ - sockaddr1

; import
align 4
@IMPORT:

library console, 'console.obj', \
        libini, 'libini.obj', \
        libio, 'libio.obj'

import  console,        \
        con_start,      'START',        \
        con_init,       'con_init',     \
        con_write_asciiz,       'con_write_asciiz',     \
        con_exit,       'con_exit',     \
        con_gets,       'con_gets',\
        con_cls,        'con_cls',\
        con_printf,     'con_printf',\
        con_getch2,     'con_getch2',\
        con_set_cursor_pos, 'con_set_cursor_pos',\
        con_set_flags,  'con_set_flags'

import  libini,         \
        ini.get_str,    'ini_get_str',\
        ini.get_int,    'ini_get_int'

import  libio,          \
        libio.init , 'lib_init'   , \
        file.size  , 'file_size'  , \
        file.open  , 'file_open'  , \
        file.read  , 'file_read'  , \
        file.close , 'file_close' , \
        file.find.first , 'file_find_first', \
        file.find.next ,  'file_find_next', \
        file.find.close , 'file_find_close'


i_end:

socketnum       dd ?


; thread specific data
socketnum2      dd ?
state           dd ?
home_dir        db '/rd/1/',0
work_dir        rb 1024
type            db ?
mode            db ?    ; active/passive

passivesocknum  dd ?
datasocketnum   dd ?

datasock:
        dw AF_INET4
.port   dw ?
.ip     dd ?
        rb 10
.length = $ - datasock

buffer  rb BUFFERSIZE
.length = $ - buffer

path    rb 1024
mem: