mte/unikernel/duniverse/mirage/lib/devices/tcp.ml
2025-11-11 02:07:51 +01:00

38 lines
1.3 KiB
OCaml

open Functoria.DSL
open Functoria.Action
type 'a tcp = TCP
type tcpv4v6 = Ip.v4v6 tcp
let tcp = Functoria.Type.Type TCP
let tcpv4v6 : tcpv4v6 typ = tcp
(* this needs to be a function due to the value restriction. *)
let tcp_direct_func () =
let packages_v = Ip.right_tcpip_library ~sublibs:[ "tcp" ] "tcpip" in
let connect _ modname = function
| [ ip ] -> code ~pos:__POS__ "%s.connect %s" modname ip
| _ -> Misc.connect_err "tcp" 1
in
impl ~packages_v ~connect "Tcp.Flow.Make" (Ip.ip @-> tcp)
let direct_tcp ip = tcp_direct_func () $ ip
let tcpv4v6_socket_conf ~ipv4_only ~ipv6_only ipv4_key ipv6_key =
let v = Runtime_arg.v in
let runtime_args = [ v ipv4_only; v ipv6_only; v ipv4_key; v ipv6_key ] in
let packages_v =
Ip.right_tcpip_library ~sublibs:[ "tcpv4v6-socket" ] "tcpip"
in
let configure i =
match Misc.get_target i with
| `Unix | `MacOSX -> ok ()
| _ -> error "TCPv4v6 socket not supported on non-UNIX targets."
in
let connect _ modname = function
| [ ipv4_only; ipv6_only; ipv4_key; ipv6_key ] ->
code ~pos:__POS__ "%s.connect ~ipv4_only:%s ~ipv6_only:%s %s %s" modname
ipv4_only ipv6_only ipv4_key ipv6_key
| _ -> Misc.connect_err "tcpv4v6_socket_conf" 4
in
impl ~packages_v ~configure ~runtime_args ~connect "Tcpv4v6_socket" tcpv4v6