module ftp /* basic ftp module RFC-959 https://tools.ietf.org/html/rfc959 Methods: ftp.connect(host) ftp.login(user, passw) pwd := ftp.pwd() ftp.cd(folder) dtp := ftp.pasv() ftp.dir() ftp.get(file) dtp.read() dtp.close() ftp.close() */ import net import io const ( connected = 220 specify_password = 331 logged_in = 230 login_first = 503 anonymous = 530 open_data_connection = 150 close_data_connection = 226 command_ok = 200 denied = 550 passive_mode = 227 complete = 226 ) struct DTP { mut: conn &net.TcpConn reader io.BufferedReader ip string port int } fn (mut dtp DTP) read() ?[]byte { mut data := []byte{} mut buf := []byte{len: 1024} for { len := dtp.reader.read(mut buf) or { break } if len == 0 { break } data << buf[..len] } return data } fn (mut dtp DTP) close() { dtp.conn.close() or { panic(err) } } struct FTP { mut: conn &net.TcpConn reader io.BufferedReader buffer_size int } pub fn new() FTP { mut f := FTP{ conn: 0 } f.buffer_size = 1024 return f } fn (mut zftp FTP) write(data string) ? { $if debug { println('FTP.v >>> $data') } zftp.conn.write('$data\r\n'.bytes()) ? } fn (mut zftp FTP) read() ?(int, string) { mut data := zftp.reader.read_line() ? $if debug { println('FTP.v <<< $data') } if data.len < 5 { return 0, '' } code := data[..3].int() if data[3] == `-` { for { data = zftp.reader.read_line() ? if data[..3].int() == code && data[3] != `-` { break } } } return code, data } pub fn (mut zftp FTP) connect(ip string) ?bool { zftp.conn = net.dial_tcp('$ip:21') ? zftp.reader = io.new_buffered_reader(reader: io.make_reader(zftp.conn)) code, _ := zftp.read() ? if code == ftp.connected { return true } return false } pub fn (mut zftp FTP) login(user string, passwd string) ?bool { zftp.write('USER $user') or { $if debug { println('ERROR sending user') } return false } mut code, _ := zftp.read() ? if code == ftp.logged_in { return true } if code != ftp.specify_password { return false } zftp.write('PASS $passwd') or { $if debug { println('ERROR sending password') } return false } code, _ = zftp.read() ? if code == ftp.logged_in { return true } return false } pub fn (mut zftp FTP) close() ? { zftp.write('QUIT') ? zftp.conn.close() ? } pub fn (mut zftp FTP) pwd() ?string { zftp.write('PWD') ? _, data := zftp.read() ? spl := data.split('"') // " if spl.len >= 2 { return spl[1] } return data } pub fn (mut zftp FTP) cd(dir string) ? { zftp.write('CWD $dir') or { return } mut code, mut data := zftp.read() ? match int(code) { ftp.denied { $if debug { println('CD $dir denied!') } } ftp.complete { code, data = zftp.read() ? } else {} } $if debug { println('CD $data') } } fn new_dtp(msg string) ?&DTP { if !is_dtp_message_valid(msg) { return error('Bad message') } ip, port := get_host_ip_from_dtp_message(msg) mut dtp := &DTP{ ip: ip port: port conn: 0 } conn := net.dial_tcp('$ip:$port') or { return error('Cannot connect to the data channel') } dtp.conn = conn dtp.reader = io.new_buffered_reader(reader: io.make_reader(dtp.conn)) return dtp } fn (mut zftp FTP) pasv() ?&DTP { zftp.write('PASV') ? code, data := zftp.read() ? $if debug { println('pass: $data') } if code != ftp.passive_mode { return error('pasive mode not allowed') } dtp := new_dtp(data) ? return dtp } pub fn (mut zftp FTP) dir() ?[]string { mut dtp := zftp.pasv() or { return error('Cannot establish data connection') } zftp.write('LIST') ? code, _ := zftp.read() ? if code == ftp.denied { return error('`LIST` denied') } if code != ftp.open_data_connection { return error('Data channel empty') } list_dir := dtp.read() ? result, _ := zftp.read() ? if result != ftp.close_data_connection { println('`LIST` not ok') } dtp.close() mut dir := []string{} sdir := list_dir.bytestr() for lfile in sdir.split('\n') { if lfile.len > 1 { spl := lfile.split(' ') dir << spl[spl.len - 1] } } return dir } pub fn (mut zftp FTP) get(file string) ?[]byte { mut dtp := zftp.pasv() or { return error('Cannot stablish data connection') } zftp.write('RETR $file') ? code, _ := zftp.read() ? if code == ftp.denied { return error('Permission denied') } if code != ftp.open_data_connection { return error('Data connection not ready') } blob := dtp.read() ? dtp.close() return blob } fn is_dtp_message_valid(msg string) bool { // An example of message: // '227 Entering Passive Mode (209,132,183,61,48,218)' return msg.contains('(') && msg.contains(')') && msg.contains(',') } fn get_host_ip_from_dtp_message(msg string) (string, int) { mut par_start_idx := -1 mut par_end_idx := -1 for i, c in msg { if c == `(` { par_start_idx = i + 1 } else if c == `)` { par_end_idx = i } } data := msg[par_start_idx..par_end_idx].split(',') ip := data[0..4].join('.') port := data[4].int() * 256 + data[5].int() return ip, port }