297 lines
12 KiB
Racket
297 lines
12 KiB
Racket
#lang racket/base
|
|
|
|
(require racket/date racket/file racket/port racket/tcp racket/list)
|
|
|
|
(provide ftp-connection?
|
|
ftp-cd
|
|
ftp-establish-connection ftp-establish-connection*
|
|
ftp-close-connection
|
|
ftp-directory-list
|
|
ftp-download-file
|
|
ftp-make-file-seconds
|
|
ftp-upload-file
|
|
ftp-delete-file
|
|
ftp-make-directory
|
|
ftp-delete-directory
|
|
ftp-rename-file)
|
|
|
|
;; opqaue record to represent an FTP connection:
|
|
(define-struct ftp-connection (in out))
|
|
|
|
(define re:multi-response-start #rx#"^[0-9][0-9][0-9]-")
|
|
(define re:response-end #rx#"^[0-9][0-9][0-9] ")
|
|
|
|
(define (check-expected-result line expected)
|
|
(when expected
|
|
(unless (ormap (lambda (expected)
|
|
(bytes=? expected (subbytes line 0 3)))
|
|
(if (bytes? expected)
|
|
(list expected)
|
|
expected))
|
|
(error 'ftp "expected result code ~a, got ~a" expected line))))
|
|
|
|
;; ftp-check-response : input-port output-port bytes-or-byteslist-or-#f (bytes any -> any) any -> any
|
|
;;
|
|
;; Checks a standard-format response, checking for the given
|
|
;; expected 3-digit result code if expected is not #f.
|
|
;;
|
|
;; While checking, the function sends response lines to
|
|
;; diagnostic-accum. This function -accum functions can return a
|
|
;; value that accumulates over multiple calls to the function, and
|
|
;; accum-start is used as the initial value. Use `void' and
|
|
;; `(void)' to ignore the response info.
|
|
;;
|
|
;; If an unexpected result is found, an exception is raised, and the
|
|
;; stream is left in an undefined state.
|
|
(define (ftp-check-response tcpin tcpout expected diagnostic-accum accum-start)
|
|
(flush-output tcpout)
|
|
(let ([line (read-bytes-line tcpin 'any)])
|
|
(cond
|
|
[(eof-object? line)
|
|
(error 'ftp "unexpected EOF")]
|
|
[(regexp-match re:multi-response-start line)
|
|
(check-expected-result line expected)
|
|
(let ([re:done (regexp (format "^~a " (subbytes line 0 3)))])
|
|
(let loop ([accum (diagnostic-accum line accum-start)])
|
|
(let ([line (read-bytes-line tcpin 'any)])
|
|
(cond [(eof-object? line)
|
|
(error 'ftp "unexpected EOF")]
|
|
[(regexp-match re:done line)
|
|
(diagnostic-accum line accum)]
|
|
[else
|
|
(loop (diagnostic-accum line accum))]))))]
|
|
[(regexp-match re:response-end line)
|
|
(check-expected-result line expected)
|
|
(diagnostic-accum line accum-start)]
|
|
[else
|
|
(error 'ftp "unexpected result: ~e" line)])))
|
|
|
|
(define (bytes->number bytes)
|
|
(string->number (bytes->string/latin-1 bytes)))
|
|
|
|
(define (get-month month-bytes)
|
|
(cond [(assoc month-bytes
|
|
'((#"Jan" 1) (#"Feb" 2) (#"Mar" 3) (#"Apr" 4) (#"May" 5)
|
|
(#"Jun" 6) (#"Jul" 7) (#"Aug" 8) (#"Sep" 9) (#"Oct" 10)
|
|
(#"Nov" 11) (#"Dec" 12)))
|
|
=> cadr]
|
|
[else (error 'get-month "bad month: ~s" month-bytes)]))
|
|
|
|
(define re:date #rx#"(...) *(.*) (..):(..)|(...) *([0-9]*) +(....)")
|
|
|
|
(define (ftp-make-file-seconds ftp-date-str)
|
|
(define date-list (regexp-match re:date (string->bytes/utf-8 ftp-date-str)))
|
|
(if (not (list-ref date-list 4))
|
|
(find-seconds 0 0 0
|
|
(bytes->number (list-ref date-list 6))
|
|
(get-month (list-ref date-list 5))
|
|
(bytes->number (list-ref date-list 7)))
|
|
(let* ([cur-secs (current-seconds)]
|
|
[cur-date (seconds->date cur-secs)]
|
|
[cur-year (date-year cur-date)]
|
|
[tzofs (date-time-zone-offset cur-date)]
|
|
[minute (bytes->number (list-ref date-list 4))]
|
|
[hour (bytes->number (list-ref date-list 3))]
|
|
[day (bytes->number (list-ref date-list 2))]
|
|
[month (get-month (list-ref date-list 1))]
|
|
[guess (+ (find-seconds 0 minute hour day month cur-year) tzofs)])
|
|
(if (guess . <= . cur-secs)
|
|
guess
|
|
(+ (find-seconds 0 minute hour day month (sub1 cur-year)) tzofs)))))
|
|
|
|
(define re:passive #rx#"\\((.*),(.*),(.*),(.*),(.*),(.*)\\)")
|
|
|
|
(define (establish-data-connection ftp-ports in-or-out)
|
|
(fprintf (ftp-connection-out ftp-ports) "PASV\r\n")
|
|
(let ([response (ftp-check-response
|
|
(ftp-connection-in ftp-ports)
|
|
(ftp-connection-out ftp-ports)
|
|
#"227" (lambda (s ignore) s) (void))])
|
|
(let* ([reg-list (regexp-match re:passive response)]
|
|
[pn1 (and reg-list
|
|
(bytes->number (list-ref reg-list 5)))]
|
|
[pn2 (bytes->number (list-ref reg-list 6))])
|
|
(unless (and reg-list pn1 pn2)
|
|
(error 'ftp "can't understand PASV response: ~e" response))
|
|
(let-values ([(tcp-data-in tcp-data-out)
|
|
(tcp-connect (format "~a.~a.~a.~a"
|
|
(list-ref reg-list 1)
|
|
(list-ref reg-list 2)
|
|
(list-ref reg-list 3)
|
|
(list-ref reg-list 4))
|
|
(+ (* 256 pn1) pn2))])
|
|
(fprintf (ftp-connection-out ftp-ports) "TYPE I\r\n")
|
|
(ftp-check-response (ftp-connection-in ftp-ports)
|
|
(ftp-connection-out ftp-ports)
|
|
#"200" void (void))
|
|
(if (eq? in-or-out 'in)
|
|
(begin
|
|
(tcp-abandon-port tcp-data-out)
|
|
tcp-data-in)
|
|
(begin
|
|
(tcp-abandon-port tcp-data-in)
|
|
tcp-data-out))))))
|
|
|
|
;; 230? is var. It always keep last line's action result. The lambda in this
|
|
;; ftp-check-response means:
|
|
;; "if one line's head is 230, then this ftp server do not
|
|
;; need PASS command. "or 230? (rege..." means if 230? is true already
|
|
;; , then do not check the line anymore, it's just true.
|
|
(define (ftp-establish-connection* in out username password)
|
|
(ftp-check-response in out #"220" void (void))
|
|
(fprintf out "USER ~a\r\n" username)
|
|
(let ([no-password? (ftp-check-response
|
|
in out (list #"331" #"230")
|
|
(lambda (line 230?)
|
|
(or 230? (regexp-match #rx#"^230" line)))
|
|
#f)])
|
|
(unless no-password?
|
|
(fprintf out "PASS ~a\r\n" password)
|
|
(ftp-check-response in out #"230" void (void))))
|
|
(make-ftp-connection in out))
|
|
|
|
(define (ftp-establish-connection server-address server-port username password)
|
|
(let-values ([(tcpin tcpout) (tcp-connect server-address server-port)])
|
|
(ftp-establish-connection* tcpin tcpout username password)))
|
|
|
|
(define (ftp-close-connection ftp-ports)
|
|
(fprintf (ftp-connection-out ftp-ports) "QUIT\r\n")
|
|
(ftp-check-response (ftp-connection-in ftp-ports)
|
|
(ftp-connection-out ftp-ports)
|
|
#"221" void (void))
|
|
(close-input-port (ftp-connection-in ftp-ports))
|
|
(close-output-port (ftp-connection-out ftp-ports)))
|
|
|
|
(define (ftp-cd ftp-ports new-dir)
|
|
(fprintf (ftp-connection-out ftp-ports) "CWD ~a\r\n" new-dir)
|
|
(ftp-check-response (ftp-connection-in ftp-ports)
|
|
(ftp-connection-out ftp-ports)
|
|
#"250" void (void)))
|
|
|
|
(define re:dir-line
|
|
(regexp (string-append
|
|
"^(.)(.*) ((?i:jan|feb|mar|apr|may|jun|jul|aug|sep|oct|nov|dec)"
|
|
" .* [0-9][0-9]:?[0-9][0-9]) (.*)$")))
|
|
|
|
(define (ftp-directory-list ftp-ports [path #f])
|
|
(define tcp-data (establish-data-connection ftp-ports 'in))
|
|
(if path
|
|
(fprintf (ftp-connection-out ftp-ports) "LIST ~a\r\n" path)
|
|
(fprintf (ftp-connection-out ftp-ports) "LIST\r\n"))
|
|
(ftp-check-response (ftp-connection-in ftp-ports)
|
|
(ftp-connection-out ftp-ports)
|
|
(list #"150" #"125") void (void))
|
|
(define lines (port->lines tcp-data))
|
|
(close-input-port tcp-data)
|
|
(ftp-check-response (ftp-connection-in ftp-ports)
|
|
(ftp-connection-out ftp-ports)
|
|
#"226" void (void))
|
|
(for*/list ([l (in-list lines)]
|
|
[m (in-value (cond [(regexp-match re:dir-line l) => cdr]
|
|
[else #f]))]
|
|
#:when m)
|
|
(define size (cond [(and (equal? "-" (car m))
|
|
(regexp-match #rx"([0-9]+) *$" (cadr m)))
|
|
=> cadr]
|
|
[else #f]))
|
|
(define r `(,(car m) ,@(cddr m)))
|
|
(if size `(,@r ,size) r)))
|
|
|
|
(define (ftp-download-file ftp-ports folder filename
|
|
#:progress [progress-proc #f])
|
|
;; Save the file under a temporary name, rename it once download is
|
|
;; complete this assures we don't over write any existing file without
|
|
;; having a good file down
|
|
(let* ([tmpfile (make-temporary-file "~a.download" #f folder)]
|
|
[new-file (open-output-file tmpfile #:exists 'truncate)]
|
|
[tcp-data (establish-data-connection ftp-ports 'in)])
|
|
|
|
(transfer-data ftp-ports 'download tcp-data new-file filename progress-proc)
|
|
|
|
(rename-file-or-directory tmpfile (build-path folder filename) #t)))
|
|
|
|
(define (ftp-upload-file ftp-ports filepath
|
|
#:progress [progress-proc #f])
|
|
(let ([upload-file (open-input-file filepath)]
|
|
[tcp-data (establish-data-connection ftp-ports 'out)])
|
|
|
|
(let ([system-type (system-path-convention-type)]
|
|
[splitter ""])
|
|
(if (eq? system-type 'unix)
|
|
(set! splitter "/")
|
|
(set! splitter "\\\\"))
|
|
|
|
(transfer-data ftp-ports 'upload upload-file tcp-data
|
|
(last (regexp-split (regexp splitter) filepath))
|
|
progress-proc))))
|
|
|
|
;; download and upload's share part
|
|
(define (transfer-data ftp-ports command from to filename progress-proc)
|
|
(let ([inner-command ""])
|
|
(cond
|
|
[(eq? command 'upload)
|
|
(set! inner-command "STOR")]
|
|
[(eq? command 'download)
|
|
(set! inner-command "RETR")])
|
|
|
|
(fprintf (ftp-connection-out ftp-ports) "~a ~a\r\n" inner-command filename))
|
|
|
|
(ftp-check-response (ftp-connection-in ftp-ports)
|
|
(ftp-connection-out ftp-ports)
|
|
(list #"125" #"150") void (void))
|
|
|
|
(let ([rcv-ch (make-channel)]
|
|
[ctrl-ch (make-channel)]
|
|
[transfer-pair (cons 0 (make-semaphore))])
|
|
|
|
(when progress-proc
|
|
(progress-proc (lambda ()
|
|
(define p transfer-pair)
|
|
(values (car p) (semaphore-peek-evt (cdr p))))))
|
|
|
|
(define bstr (make-bytes 40960))
|
|
(let loop ()
|
|
(let ([n (read-bytes! bstr from)])
|
|
(unless (eof-object? n)
|
|
(define sent (write-bytes bstr to 0 n))
|
|
(when progress-proc
|
|
(define old-pair transfer-pair)
|
|
(set! transfer-pair (cons (+ sent (car old-pair))
|
|
(make-semaphore)))
|
|
(semaphore-post (cdr old-pair)))
|
|
(loop)))))
|
|
|
|
(close-input-port from)
|
|
(close-output-port to)
|
|
(ftp-check-response (ftp-connection-in ftp-ports)
|
|
(ftp-connection-out ftp-ports)
|
|
#"226" void (void)))
|
|
|
|
(define (ftp-delete-file ftp-ports filepath)
|
|
(fprintf (ftp-connection-out ftp-ports) "DELE ~a\r\n" filepath)
|
|
(ftp-check-response (ftp-connection-in ftp-ports)
|
|
(ftp-connection-out ftp-ports)
|
|
#"250" void (void)))
|
|
|
|
(define (ftp-make-directory ftp-ports dirname)
|
|
(fprintf (ftp-connection-out ftp-ports) "MKD ~a\r\n" dirname)
|
|
(ftp-check-response (ftp-connection-in ftp-ports)
|
|
(ftp-connection-out ftp-ports)
|
|
#"257" void (void)))
|
|
|
|
(define (ftp-delete-directory ftp-ports dirname)
|
|
(fprintf (ftp-connection-out ftp-ports) "RMD ~a\r\n" dirname)
|
|
(ftp-check-response (ftp-connection-in ftp-ports)
|
|
(ftp-connection-out ftp-ports)
|
|
#"250" void (void)))
|
|
|
|
(define (ftp-rename-file ftp-ports origin dest)
|
|
(fprintf (ftp-connection-out ftp-ports) "RNFR ~a\r\n" origin)
|
|
(ftp-check-response (ftp-connection-in ftp-ports)
|
|
(ftp-connection-out ftp-ports)
|
|
#"350" void (void))
|
|
(fprintf (ftp-connection-out ftp-ports) "RNTO ~a\r\n" dest)
|
|
(ftp-check-response (ftp-connection-in ftp-ports)
|
|
(ftp-connection-out ftp-ports)
|
|
#"250" void (void)))
|