1

我正在尝试编写一个小型 Lwt（和电池）端口扫描仪以更好地理解 Lwt，但是每当我尝试使用 scan_ports_range 一次扫描太多端口时，我都会遇到一个奇怪的异常。我怀疑我的保持最大连接打开数的机制不起作用......

Exception: Unix.Unix_error (Batteries.Unix.EMFILE, "socket",""). Fatal error: exception Sys_error("/home/(censored)/.opam/4.00.1/lib/utop: Too many open files") （顺便说一句，这也会崩溃）

代码如下。要触发错误评估： scan_ports_range ~host:"127.0.0.1" (1000,2000)

也欢迎对我的 lwt 风格提出任何批评/建议，因为我才刚刚开始学习它。

open Lwt

let addr_parts addr = 
  let (host, port) = String.split addr ":" in (host, int_of_string port)

let addr ~host ~port = 
  lwt entry = Lwt_unix.gethostbyname host in
  if Array.length entry.Unix.h_addr_list = 0 then begin
    failwith (Printf.sprintf "no address found for host %S\n" host)
  end;
  return (Unix.ADDR_INET (entry.Unix.h_addr_list.(0), port))

let test_connection ?(timeout=1.0) addr = 
  let fd = Lwt_unix.socket Unix.PF_INET Unix.SOCK_STREAM 0 in
  let connect_close = 
    (Lwt_unix.connect fd addr) >>= (fun () -> Lwt_unix.close fd)  in
  try_lwt
    (pick [connect_close ; Lwt_unix.timeout timeout])
    >>= (fun () -> return true)
  with _ -> return false

let scan_ports ~host ~ports =
  ports |> Lwt_list.map_p (fun port ->
        lwt adr = addr ~host ~port in
        test_connection adr >>= (fun res -> return (res,port)) )
    >>= (fun l -> return ( l |> List.filter_map ( function
      | false, _ -> None | true, port -> Some(port) ) ) )

let scan_ports_range ?(max_open=20) ~host (a, b) =
  let rec loop acc enum = 
    match Enum.peek enum with
    | None -> acc |> List.concat |> List.rev |> return
    | Some _ -> 
        let ports = enum |> Enum.take max_open |> List.of_enum in
        let open_ports = scan_ports ~host ~ports in
        open_ports >>= (fun l -> loop (l::acc) enum )
  in loop [] (a--b)
4

2 回答 2

6

作为一个疯狂的猜测，我认为你需要在超时的情况下强制关闭套接字，所以 cullprint 可能是：

pick [connect_close ; Lwt_unix.timeout timeout]
于 2012-11-09T08:19:15.203 回答
1

这是根据 Thomas 的更正代码。

let test_connection ?(timeout=1.0) addr = 
  let fd = Lwt_unix.socket Unix.PF_INET Unix.SOCK_STREAM 0 in
  let connect_close = 
    (Lwt_unix.connect fd addr) >>= (fun () -> Lwt_unix.close fd)  in
  try_lwt
    (pick [connect_close ; Lwt_unix.timeout timeout])
    >>= (fun () -> return true)
  with _ -> (Lwt_unix.close fd) >>= (fun () -> return false)
于 2012-11-09T23:17:32.497 回答