Source file mirage_impl_resolver.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
open Functoria
module Key = Mirage_key
open Mirage_impl_misc
open Mirage_impl_mclock
open Mirage_impl_pclock
open Mirage_impl_stack
open Mirage_impl_random
open Mirage_impl_time

type resolver = Resolver

let resolver = Type.v Resolver

let resolver_unix_system =
  let packages_v =
    Key.(if_ is_unix)
      [
        Mirage_impl_conduit.pkg;
        package ~min:"4.0.0" ~max:"6.0.0" "conduit-lwt-unix";
      ]
      []
  in
  let configure i =
    match get_target i with
    | `Unix | `MacOSX -> Action.ok ()
    | _ -> Action.error "Unix resolver not supported on non-UNIX targets."
  in
  let connect _ _modname _ = "Lwt.return Resolver_lwt_unix.system" in
  impl ~packages_v ~configure ~connect "Resolver_lwt" resolver

let resolver_dns_conf ~ns ~ns_port =
  let packages = [ Mirage_impl_conduit.pkg ] in
  let keys = Key.[ v ns; v ns_port ] in
  let connect _ modname = function
    | [ _r; _t; _m; _p; stack ] ->
        Fmt.str
          "let nameservers = Option.map (fun x -> [ `Plaintext (x, %a) ]) %a \
           in@;\
           let res = %s.v ?nameservers %s in@;\
           Lwt.return res@;"
          pp_key ns_port pp_key ns modname stack
    | _ -> failwith (connect_err "resolver" 3)
  in
  impl ~packages ~keys ~connect "Resolver_mirage.Make"
    (random @-> time @-> mclock @-> pclock @-> stackv4v6 @-> resolver)

let resolver_dns ?ns ?ns_port ?(time = default_time)
    ?(mclock = default_monotonic_clock) ?(pclock = default_posix_clock)
    ?(random = default_random) stack =
  let ns = Key.resolver ?default:ns ()
  and ns_port = Key.resolver_port ?default:ns_port () in
  resolver_dns_conf ~ns ~ns_port $ random $ time $ mclock $ pclock $ stack