293 lines
7.8 KiB
PHP
293 lines
7.8 KiB
PHP
server_parser:
|
|
|
|
; Commands are always 3 numbers and followed by a space
|
|
; If a server decides it needs multiline output,
|
|
; first lines will have a dash instead of space after numbers,
|
|
; thus they are simply ignored in this simple command parser.
|
|
|
|
cmp dword[buf_cmd+4], "Open"
|
|
je init_download_count
|
|
|
|
cmp dword[buf_cmd], "150 "
|
|
je data_loop
|
|
|
|
cmp dword[buf_cmd], "220 "
|
|
je connect_ok
|
|
|
|
; cmp dword[buf_cmd], "226 "
|
|
; je transfer_ok
|
|
|
|
cmp dword[buf_cmd], "227 "
|
|
je pasv_ok
|
|
|
|
cmp dword[buf_cmd], "230 "
|
|
je login_ok
|
|
|
|
; cmp dword[buf_cmd], "250"
|
|
; je op_ok
|
|
|
|
cmp dword[buf_cmd], "331 "
|
|
je pass
|
|
|
|
; cmp dword[buf_cmd], "421 "
|
|
; je timeout
|
|
|
|
cmp dword[buf_cmd], "503 " ; login first
|
|
je welcome
|
|
|
|
cmp dword[buf_cmd], "530 " ; password incorrect
|
|
je welcome
|
|
|
|
cmp dword[buf_cmd], "550 "
|
|
je close_datacon
|
|
|
|
cmp byte[buf_cmd+3], "-"
|
|
je wait_for_servercommand
|
|
jmp wait_for_usercommand
|
|
|
|
|
|
welcome:
|
|
|
|
mov [status], STATUS_CONNECTED
|
|
jmp wait_for_usercommand
|
|
|
|
|
|
connect_ok:
|
|
|
|
mov [status], STATUS_CONNECTED
|
|
jmp arg_handler.copy_user
|
|
|
|
pass:
|
|
|
|
mov [status], STATUS_NEEDPASSWORD
|
|
jmp wait_for_usercommand
|
|
|
|
|
|
login_ok:
|
|
|
|
mov [status], STATUS_LOGGED_IN
|
|
|
|
cmp [param_path], 0x20 ; no path specified
|
|
jbe wait_for_usercommand
|
|
|
|
; copy path to buf_cmd and execute CWD
|
|
jmp arg_handler.get_path
|
|
|
|
|
|
pasv_ok:
|
|
|
|
sub ecx, 4
|
|
jb .fail
|
|
mov al, "("
|
|
mov edi, buf_cmd + 4
|
|
repne scasb
|
|
|
|
mcall socket, AF_INET4, SOCK_STREAM, 0
|
|
cmp eax, -1
|
|
jne @f
|
|
mov eax, str_err_socket
|
|
jmp error
|
|
@@: mov [datasocket], eax
|
|
|
|
mov esi, edi
|
|
call ascii_dec
|
|
mov byte[sockaddr2.ip+0], bl
|
|
call ascii_dec
|
|
mov byte[sockaddr2.ip+1], bl
|
|
call ascii_dec
|
|
mov byte[sockaddr2.ip+2], bl
|
|
call ascii_dec
|
|
mov byte[sockaddr2.ip+3], bl
|
|
|
|
call ascii_dec
|
|
mov byte[sockaddr2.port+0], bl
|
|
call ascii_dec
|
|
mov byte[sockaddr2.port+1], bl
|
|
|
|
icall eax, interface_addr, interface.print, str_open
|
|
mcall connect, [datasocket], sockaddr2, 18
|
|
cmp eax, -1
|
|
jne @f
|
|
mov eax, str_err_connect
|
|
jmp error
|
|
@@: jmp wait_for_servercommand
|
|
|
|
.fail:
|
|
icall eax, interface_addr, interface.print, str_unknown
|
|
jmp wait_for_servercommand
|
|
|
|
|
|
; get file size, initialize count for number of bytes downloaded
|
|
init_download_count:
|
|
|
|
mov edx, 0
|
|
; search for 'Open' in buf_cmd
|
|
lea esi, [buf_cmd+3]
|
|
@@:
|
|
inc esi
|
|
cmp dword[esi], 'Open'
|
|
je @f
|
|
cmp byte[esi], 0
|
|
jne @b
|
|
jmp data_loop
|
|
|
|
@@:
|
|
; get file size
|
|
mov al, '('
|
|
mov ecx, -1
|
|
mov edi, buf_cmd
|
|
repne scasb
|
|
xor eax, eax
|
|
mov ebx, 10
|
|
xor ecx, ecx
|
|
mov esi, edi
|
|
@@:
|
|
push eax
|
|
lodsb
|
|
sub al, '0'
|
|
mov cl, al
|
|
pop eax
|
|
mul ebx
|
|
add eax, ecx
|
|
cmp byte[esi], ' '
|
|
jne @b
|
|
mov [file_size], eax
|
|
|
|
data_loop:
|
|
|
|
cmp [operation], OPERATION_STOR
|
|
je .stor
|
|
|
|
push edx
|
|
; we are receiving data
|
|
mcall recv, [datasocket], buf_buffer2, BUFFERSIZE, 0
|
|
pop edx ; get byte count
|
|
add edx, eax
|
|
test ebx, ebx
|
|
jnz .done
|
|
mov byte[buf_buffer2 + eax], 0
|
|
|
|
cmp [operation], OPERATION_RETR
|
|
je .retr
|
|
|
|
cmp [operation], OPERATION_RDIR
|
|
je .rdir
|
|
|
|
cmp [operation], OPERATION_LIST
|
|
je .list
|
|
|
|
; not retreiving, just print to console
|
|
icall eax, interface_addr, interface.print, buf_buffer2
|
|
jmp data_loop
|
|
|
|
; for console, simply print. for gui, add name to tree list
|
|
.list:
|
|
ijmp ebx, interface_addr, interface.list
|
|
|
|
|
|
; retreiving, save to file
|
|
.retr:
|
|
mov [filestruct.ptr], buf_buffer2
|
|
mov [filestruct.size], eax
|
|
push eax
|
|
mcall 70, filestruct
|
|
test eax, eax
|
|
jz @f
|
|
call error_fs
|
|
jmp close_datacon
|
|
@@:
|
|
pop eax
|
|
add [filestruct.offset], eax
|
|
|
|
icall eax, interface_addr, interface.progress
|
|
jmp data_loop
|
|
|
|
; storing, send all data
|
|
.stor:
|
|
mcall 70, filestruct
|
|
cmp eax, 6 ; end of file
|
|
je .last_call
|
|
test eax, eax ; error
|
|
jz @f
|
|
call error_fs
|
|
jmp close_datacon
|
|
@@:
|
|
add [filestruct.offset], ebx
|
|
mov esi, ebx
|
|
mcall send, [datasocket], buf_buffer2, , 0
|
|
mov edx, [filestruct.offset]
|
|
icall eax, interface_addr, interface.progress
|
|
jmp .stor
|
|
|
|
.last_call:
|
|
mov esi, ebx
|
|
mcall send, [datasocket], buf_buffer2, , 0
|
|
mov edx, [filestruct.offset]
|
|
icall eax, interface_addr, interface.progress
|
|
|
|
.done:
|
|
icall eax, interface_addr, interface.print, str_close
|
|
mcall close, [datasocket]
|
|
mov [operation], OPERATION_NONE
|
|
jmp wait_for_servercommand
|
|
|
|
.rdir:
|
|
; alloc/realloc memory block to store filenames
|
|
mov ecx, eax ; eax is size of buffer received
|
|
inc ecx
|
|
add ecx, [size_fname] ; added old size to form new required size
|
|
mcall 68, 20, , [ptr_fname] ; realloc
|
|
test eax, eax
|
|
je error_heap
|
|
mov [ptr_fname], eax ; eax contains the new block now
|
|
mov [ptr_queue], eax
|
|
|
|
; copy filenames into fname buffer
|
|
mov esi, buf_buffer2
|
|
mov edi, eax
|
|
add edi, [size_fname]
|
|
.copy_buf:
|
|
lodsb
|
|
cmp al, 13 ; ignore any carriage return character
|
|
je .copy_buf
|
|
stosb
|
|
cmp al, 10 ; linefeed marks end of filename
|
|
je @f
|
|
inc [queued]
|
|
@@:
|
|
test al, al ; 0 marks end of buffer
|
|
jne .copy_buf
|
|
|
|
; All received filenames have been copied, calculate new size of fname buffer
|
|
dec edi ; dont count the trailing 0 byte
|
|
sub edi, [ptr_fname]
|
|
mov [size_fname], edi
|
|
jmp data_loop
|
|
|
|
|
|
close_datacon:
|
|
cmp [operation], OPERATION_NONE
|
|
je wait_for_usercommand
|
|
icall eax, interface_addr, interface.print, str_close
|
|
mcall close, [datasocket]
|
|
jmp wait_for_usercommand
|
|
|
|
|
|
ascii_dec:
|
|
|
|
xor ebx, ebx
|
|
mov cl, 4 ; max length is 3 digits + 1 separator
|
|
.loop:
|
|
lodsb
|
|
sub al, '0'
|
|
jb .done
|
|
cmp al, 9
|
|
ja .done
|
|
lea ebx, [ebx*4+ebx] ; ebx *5
|
|
shl ebx, 1 ; ebx *2
|
|
add bl, al
|
|
dec cl
|
|
jnz .loop
|
|
|
|
.done:
|
|
ret |