forked from MLstate/opalang
-
Notifications
You must be signed in to change notification settings - Fork 0
/
ftpClient.ml
106 lines (93 loc) · 4.34 KB
/
ftpClient.ml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
(*
Copyright © 2011 MLstate
This file is part of OPA.
OPA is free software: you can redistribute it and/or modify it under the
terms of the GNU Affero General Public License, version 3, as published by
the Free Software Foundation.
OPA is distributed in the hope that it will be useful, but WITHOUT ANY
WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
FOR A PARTICULAR PURPOSE. See the GNU Affero General Public License for
more details.
You should have received a copy of the GNU Affero General Public License
along with OPA. If not, see <http://www.gnu.org/licenses/>.
*)
(* ftpClient: Simple read of file from FTP server.
*)
#<Debugvar:PROTOCOL_DEBUG>
module FCC = FtpClientCore
module List = Base.List
module String = Base.String
module Rc = Rcontent
let protocol = FCC.protocol
exception Bad_address of string
let _log sep code reason = Logger.warning "%d%s%s" code sep reason
let get_match str n = try (Some (Str.matched_group n str)) with Not_found -> None
let find_matches str cnt = let rec aux n l = if n > cnt then l else aux (n+1) (l@[get_match str n]) in aux 1 []
let ftpre =
Str.regexp
"\\(ftp://\\)?\\(\\([a-zA-Z0-9_-\\.]+\\)\\(:\\([a-zA-Z0-9_-\\.]+\\)\\)?@\\)?\\([a-zA-Z0-9_-\\.]+\\):?\\([0-9]+\\)?/\\(.*\\)"
let split_ftp user pass port s =
if Str.string_match ftpre s 0
then
match (get_match s 3, get_match s 5, get_match s 6, get_match s 7, get_match s 8) with
| (user_opt, pass_opt, Some domain, port_opt, Some path) ->
let port =
match port_opt with
| Some portstr -> (try int_of_string portstr with Failure "int_of_string" -> port)
| None -> port in
let user = Option.default user user_opt in
let pass = Option.default pass pass_opt in
(user, pass, domain, port, path)
| _ -> raise (Bad_address s)
else
raise (Bad_address s)
let get_ip name = (Unix.gethostbyname name).Unix.h_addr_list.(0)
let my_ip () = Unix.string_of_inet_addr(get_ip (Unix.gethostname()))
let dre = Str.regexp_string "."
let dataport () =
match Str.split dre (my_ip ()) with
| [h1;h2;h3;h4] ->
let dp = Random.int (65534-49152) + 49152 in
let p1, p2 = (dp land 0xff00) lsr 8, dp land 0xff in
let addr = Unix.inet_addr_of_string (Printf.sprintf "%s.%s.%s.%s" h1 h2 h3 h4) in
#<If$minlevel 2>Logger.debug "dataport: addr=%s\n" (Unix.string_of_inet_addr addr)#<End>;
let port_spec = Network.make_port_spec ~protocol addr dp in
port_spec, Printf.sprintf "%s,%s,%s,%s,%d,%d" h1 h2 h3 h4 p1 p2
| _ -> assert false
let receive_ftp_file sched url
?(user="anonymous") ?(password="change.me@example.com") ?(ct=Rc.CT_BUFFER) ?(port=21)
?(datablocksize=4096) ?(hint=4096)
cont =
let user, password, domain, port, filespec = split_ftp user password port url in
let pathname = Filename.dirname filespec in
let filename = Filename.basename filespec in
let dataportspec, dataportstr = dataport () in
let datasecuremode = Network.Unsecured in
#<If>Logger.debug "receive_ftp_file: domain=%s port=%d pathname=%s filename=%s dataportstr=%s\n"
domain port pathname filename dataportstr#<End>;
let ftp = { FCC.content_type=ct; content_hint=hint;
user=user; password=password;
domain=domain; pathname=pathname; filename=filename;
dataportstr=dataportstr; datasecuremode=datasecuremode;
dataportspec=dataportspec; datablocksize=datablocksize;
} in
let state = {
FCC.log = _log " ";
elog = _log "-";
cont = (function
| FCC.Error msg ->
(Logger.error "ERROR: %s\n" msg;
cont (FCC.Error msg))
| FCC.Ok res -> cont (FCC.Ok res));
} in
let client = { FCC.runtime = { FCC.rt_plim = 128;
rt_buf = Buffer.create 0;
rt_proto = { FCC.rt_block_size = 4096;
rt_backtrace = true;
rt_server_write_timeout = Time.seconds 36;
rt_payload = ();
};
};
err_cont = None;
extra_params = (state,ftp) } in
FCC.connect client sched domain port