/
githubmirror
/
deno
Обзор
Документация
Войти
/
githubmirror
/
deno
Код
Запросы
0
Пакеты
0
Релизы
0
Аналитика
Безопасность
main
ext/net/ops.rs
1 747 строк
47 KB
Nathan Whitaker
fix(net): handle malformed DNS record text (#36374)
05 авг 2026, 00:46
Не верифицирован
05 авг 2026, 00:46
b82aa41
Код
Авторство
О чём код?
// Copyright 2018-2026 the Deno authors. MIT license. use std::borrow::Cow; use std::cell::RefCell; use std::net::Ipv4Addr; use std::net::Ipv6Addr; use std::net::SocketAddr; use std::rc::Rc; use std::str::FromStr; use std::time::Duration; use deno_core::AsyncRefCell; use deno_core::ByteString; use deno_core::CancelFuture; use deno_core::CancelHandle; use deno_core::CancelTryFuture; use deno_core::FromV8; use deno_core::JsBuffer; use deno_core::OpState; use deno_core::RcRef; use deno_core::Resource; use deno_core::ResourceId; use deno_core::ToV8; use deno_core::op2; use deno_permissions::PermissionsContainer; use hickory_proto::ProtoError; use hickory_proto::ProtoErrorKind; use hickory_proto::rr::Record; use hickory_proto::rr::record_data::RData; use hickory_proto::rr::record_type::RecordType; use hickory_resolver::ResolveError; use hickory_resolver::ResolveErrorKind; use hickory_resolver::config::NameServerConfigGroup; use hickory_resolver::config::ResolverConfig; use hickory_resolver::config::ResolverOpts; use hickory_resolver::name_server::TokioConnectionProvider; use hickory_resolver::system_conf; use quinn::rustls; use serde::Deserialize; use serde::Serialize; use socket2::Domain; use socket2::Protocol; use socket2::Socket; use socket2::Type; use tokio::net::TcpStream; use tokio::net::UdpSocket; use crate::io::TcpStreamResource; use crate::raw::NetworkListenerResource; use crate::resolve_addr::resolve_addr; use crate::resolve_addr::resolve_addr_sync; use crate::tcp::TcpListener; use crate::tunnel::TunnelAddr; pub type Fd = u32; #[derive(Serialize, Clone, Debug)] #[serde(rename_all = "camelCase")] pub struct TlsHandshakeInfo { pub alpn_protocol: Option<ByteString>, #[serde(skip_serializing)] pub peer_certificates: Option<Vec<rustls::pki_types::CertificateDer<'static>>>, } #[derive(Debug, FromV8, ToV8)] pub struct IpAddr { pub hostname: String, pub port: u16, } impl From<SocketAddr> for IpAddr { fn from(addr: SocketAddr) -> Self { Self { hostname: addr.ip().to_string(), port: addr.port(), } } } impl From<TunnelAddr> for IpAddr { fn from(addr: TunnelAddr) -> Self { Self { hostname: addr.hostname(), port: addr.port(), } } } /// Options for TCP connection (used by Deno.connect and node:net) #[derive(Debug, Deserialize, Default)] #[serde(rename_all = "camelCase")] pub struct TcpConnectOptions { /// Enable Happy Eyeballs (auto select family). Default: true #[serde(default = "default_auto_select_family")] pub auto_select_family: bool, /// Delay in milliseconds between connection attempts. Default: 250 #[serde(default = "default_attempt_delay")] pub auto_select_family_attempt_delay: u64, } fn default_auto_select_family() -> bool { true } fn default_attempt_delay() -> u64 { crate::happy_eyeballs::DEFAULT_ATTEMPT_DELAY_MS } #[derive(Debug, thiserror::Error, deno_error::JsError)] pub enum NetError { #[class("BadResource")] #[error("Listener has been closed")] ListenerClosed, #[class("Busy")] #[error("Listener already in use")] ListenerBusy, #[class("BadResource")] #[error("Socket has been closed")] SocketClosed, #[class("NotConnected")] #[error("Socket has been closed")] SocketClosedNotConnected, #[class("Busy")] #[error("Socket already in use")] SocketBusy, #[class(inherit)] #[error("{0}")] Io(#[from] std::io::Error), #[class("Busy")] #[error("Another accept task is ongoing")] AcceptTaskOngoing, #[class(inherit)] #[error(transparent)] Permission(#[from] deno_permissions::PermissionCheckError), #[class(inherit)] #[error("{0}")] Resource(#[from] deno_core::error::ResourceError), #[class(generic)] #[error("No resolved address found")] NoResolvedAddress, #[class(generic)] #[error("{0}")] AddrParse(#[from] std::net::AddrParseError), #[class(inherit)] #[error("{0}")] Map(crate::io::MapError), #[class(inherit)] #[error("{0}")] Canceled(#[from] deno_core::Canceled), #[class("NotFound")] #[error("{0}")] DnsNotFound(ResolveError), #[class("NotConnected")] #[error("{0}")] DnsNotConnected(ResolveError), #[class("TimedOut")] #[error("{0}")] DnsTimedOut(ResolveError), #[class(generic)] #[error("{0}")] Dns(#[from] ResolveError), #[class("NotSupported")] #[error("Provided record type is not supported")] UnsupportedRecordType, #[class("InvalidData")] #[error("File name or path {0:?} is not valid UTF-8")] InvalidUtf8(std::ffi::OsString), #[class(generic)] #[error("unexpected key type")] UnexpectedKeyType, #[class(type)] #[error("Invalid hostname: '{0}'")] InvalidHostname(String), #[class("Busy")] #[error("TCP stream is currently in use")] TcpStreamBusy, #[class(generic)] #[error("{0}")] Rustls(#[from] deno_tls::rustls::Error), #[class(inherit)] #[error("{0}")] Tls(#[from] deno_tls::TlsError), #[class("InvalidData")] #[error("Error creating TLS certificate: Deno.listenTls requires a key")] ListenTlsRequiresKey, #[class(inherit)] #[error("{0}")] RootCertStore(deno_error::JsErrorBox), #[class(generic)] #[error("{0}")] Reunite(tokio::net::tcp::ReuniteError), #[class(generic)] #[error("VSOCK is not supported on this platform")] VsockUnsupported, #[class(generic)] #[error("Tunnel is not open")] TunnelMissing, } pub(crate) fn accept_err(e: std::io::Error) -> NetError { if let std::io::ErrorKind::Interrupted = e.kind() { NetError::ListenerClosed } else { NetError::Io(e) } } #[op2] pub async fn op_net_accept_tcp( state: Rc<RefCell<OpState>>, #[smi] rid: ResourceId, ) -> Result<(ResourceId, IpAddr, IpAddr, Option<Fd>), NetError> { let resource = state .borrow() .resource_table .get::<NetworkListenerResource<TcpListener>>(rid) .map_err(|_| NetError::ListenerClosed)?; let listener = RcRef::map(&resource, |r| &r.listener) .try_borrow_mut() .ok_or_else(|| NetError::AcceptTaskOngoing)?; let cancel = RcRef::map(resource, |r| &r.cancel); let (tcp_stream, _socket_addr) = listener .accept() .try_or_cancel(cancel) .await .map_err(accept_err)?; let mut _fd_raw: Option<Fd> = None; #[cfg(not(windows))] { use std::os::fd::AsFd; use std::os::fd::AsRawFd; let fd = tcp_stream.as_fd(); _fd_raw = Some(fd.as_raw_fd() as u32); } let local_addr = tcp_stream.local_addr()?; let remote_addr = tcp_stream.peer_addr()?; let mut state = state.borrow_mut(); let rid = state .resource_table .add(TcpStreamResource::new(tcp_stream.into_split())); Ok(( rid, IpAddr::from(local_addr), IpAddr::from(remote_addr), _fd_raw, )) } #[op2] pub async fn op_net_recv_udp( state: Rc<RefCell<OpState>>, #[smi] rid: ResourceId, #[buffer] mut buf: JsBuffer, ) -> Result<(usize, IpAddr), NetError> { let resource = state .borrow_mut() .resource_table .get::<UdpSocketResource>(rid) .map_err(|_| NetError::SocketClosed)?; let socket = RcRef::map(&resource, |r| &r.socket).borrow().await; let cancel_handle = RcRef::map(&resource, |r| &r.cancel); let (nread, remote_addr) = socket .recv_from(&mut buf) .try_or_cancel(cancel_handle) .await?; Ok((nread, IpAddr::from(remote_addr))) } #[op2(stack_trace)] #[number] pub async fn op_net_send_udp( state: Rc<RefCell<OpState>>, #[smi] rid: ResourceId, #[scoped] addr: IpAddr, #[buffer] zero_copy: JsBuffer, ) -> Result<usize, NetError> { { let mut s = state.borrow_mut(); s.borrow_mut::<PermissionsContainer>().check_net( &(&addr.hostname, Some(addr.port)), "Deno.DatagramConn.send()", )?; } let addr = resolve_addr(&addr.hostname, addr.port) .await? .next() .ok_or(NetError::NoResolvedAddress)?; { state .borrow_mut() .borrow_mut::<PermissionsContainer>() .check_net_resolved( &addr.ip(), addr.port(), "Deno.DatagramConn.send()", )?; } let resource = state .borrow_mut() .resource_table .get::<UdpSocketResource>(rid) .map_err(|_| NetError::SocketClosed)?; let socket = RcRef::map(&resource, |r| &r.socket).borrow().await; let nwritten = socket.send_to(&zero_copy, &addr).await?; Ok(nwritten) } #[op2(fast)] pub fn op_net_validate_multicast( #[string] address: String, #[string] multi_interface: String, ) -> Result<(), NetError> { let addr = Ipv4Addr::from_str(address.as_str())?; let interface_addr = Ipv4Addr::from_str(multi_interface.as_str())?; if !addr.is_multicast() { return Err(NetError::InvalidHostname(address)); } if !interface_addr.is_multicast() { return Err(NetError::InvalidHostname(multi_interface)); } Ok(()) } #[op2] pub async fn op_net_join_multi_v4_udp( state: Rc<RefCell<OpState>>, #[smi] rid: ResourceId, #[string] address: String, #[string] multi_interface: String, ) -> Result<(), NetError> { let resource = state .borrow_mut() .resource_table .get::<UdpSocketResource>(rid) .map_err(|_| NetError::SocketClosed)?; let socket = RcRef::map(&resource, |r| &r.socket).borrow().await; let addr = Ipv4Addr::from_str(address.as_str())?; let interface_addr = Ipv4Addr::from_str(multi_interface.as_str())?; socket.join_multicast_v4(addr, interface_addr)?; Ok(()) } #[op2] pub async fn op_net_join_multi_v6_udp( state: Rc<RefCell<OpState>>, #[smi] rid: ResourceId, #[string] address: String, #[smi] multi_interface: u32, ) -> Result<(), NetError> { let resource = state .borrow_mut() .resource_table .get::<UdpSocketResource>(rid) .map_err(|_| NetError::SocketClosed)?; let socket = RcRef::map(&resource, |r| &r.socket).borrow().await; let addr = Ipv6Addr::from_str(address.as_str())?; socket.join_multicast_v6(&addr, multi_interface)?; Ok(()) } #[op2] pub async fn op_net_leave_multi_v4_udp( state: Rc<RefCell<OpState>>, #[smi] rid: ResourceId, #[string] address: String, #[string] multi_interface: String, ) -> Result<(), NetError> { let resource = state .borrow_mut() .resource_table .get::<UdpSocketResource>(rid) .map_err(|_| NetError::SocketClosed)?; let socket = RcRef::map(&resource, |r| &r.socket).borrow().await; let addr = Ipv4Addr::from_str(address.as_str())?; let interface_addr = Ipv4Addr::from_str(multi_interface.as_str())?; socket.leave_multicast_v4(addr, interface_addr)?; Ok(()) } #[op2] pub async fn op_net_leave_multi_v6_udp( state: Rc<RefCell<OpState>>, #[smi] rid: ResourceId, #[string] address: String, #[smi] multi_interface: u32, ) -> Result<(), NetError> { let resource = state .borrow_mut() .resource_table .get::<UdpSocketResource>(rid) .map_err(|_| NetError::SocketClosed)?; let socket = RcRef::map(&resource, |r| &r.socket).borrow().await; let addr = Ipv6Addr::from_str(address.as_str())?; socket.leave_multicast_v6(&addr, multi_interface)?; Ok(()) } #[op2] pub async fn op_net_set_multi_loopback_udp( state: Rc<RefCell<OpState>>, #[smi] rid: ResourceId, is_v4_membership: bool, loopback: bool, ) -> Result<(), NetError> { let resource = state .borrow_mut() .resource_table .get::<UdpSocketResource>(rid) .map_err(|_| NetError::SocketClosed)?; let socket = RcRef::map(&resource, |r| &r.socket).borrow().await; if is_v4_membership { socket.set_multicast_loop_v4(loopback)?; } else { socket.set_multicast_loop_v6(loopback)?; } Ok(()) } #[op2] pub async fn op_net_set_multi_ttl_udp( state: Rc<RefCell<OpState>>, #[smi] rid: ResourceId, #[smi] ttl: u32, ) -> Result<(), NetError> { let resource = state .borrow_mut() .resource_table .get::<UdpSocketResource>(rid) .map_err(|_| NetError::SocketClosed)?; let socket = RcRef::map(&resource, |r| &r.socket).borrow().await; socket.set_multicast_ttl_v4(ttl)?; Ok(()) } #[op2] pub async fn op_net_set_broadcast_udp( state: Rc<RefCell<OpState>>, #[smi] rid: ResourceId, broadcast: bool, ) -> Result<(), NetError> { let resource = state .borrow_mut() .resource_table .get::<UdpSocketResource>(rid) .map_err(|_| NetError::SocketClosed)?; let socket = RcRef::map(&resource, |r| &r.socket).borrow().await; socket.set_broadcast(broadcast)?; Ok(()) } /// If this token is present in op_net_connect_tcp call and /// the hostname matches with one of the resolved IPs, then /// the permission check is performed against the original hostname. pub struct NetPermToken { pub hostname: String, pub port: Option<u16>, pub resolved_ips: Vec<String>, } // SAFETY: we're sure `NetPermToken` can be GCed unsafe impl deno_core::GarbageCollected for NetPermToken { fn trace(&self, _visitor: &mut deno_core::v8::cppgc::Visitor) {} fn get_name(&self) -> &'static std::ffi::CStr { c"NetPermToken" } } impl NetPermToken { /// Checks if the given address is included in the resolved IPs. pub fn includes(&self, addr: &str) -> bool { self.resolved_ips.iter().any(|ip| ip == addr) } /// Returns the host that `--allow-net` should be checked against for a /// connection to `address`: the token's original hostname when `address` is /// one of the IPs this token resolved, otherwise the literal `address`. This /// keeps the hostname from being grafted onto an unrelated IP. pub fn check_host<'a>(&'a self, address: &'a str) -> &'a str { if self.includes(address) { &self.hostname } else { address } } } #[op2] pub fn op_net_get_ips_from_perm_token( #[cppgc] token: &NetPermToken, ) -> Vec<String> { token.resolved_ips.clone() } #[op2(stack_trace)] pub async fn op_net_connect_tcp( state: Rc<RefCell<OpState>>, #[scoped] addr: IpAddr, #[cppgc] net_perm_token: Option<&NetPermToken>, #[smi] resource_abort_id: Option<ResourceId>, #[serde] options: Option<TcpConnectOptions>, ) -> Result<(ResourceId, IpAddr, IpAddr), NetError> { op_net_connect_tcp_inner( state, addr, net_perm_token, resource_abort_id, options, ) .await } #[inline] pub async fn op_net_connect_tcp_inner( state: Rc<RefCell<OpState>>, addr: IpAddr, net_perm_token: Option<&NetPermToken>, resource_abort_id: Option<ResourceId>, options: Option<TcpConnectOptions>, ) -> Result<(ResourceId, IpAddr, IpAddr), NetError> { { let mut state_ = state.borrow_mut(); // If token exists and the address matches to its resolved ips, // then we can check net permission against token.hostname, instead of addr.hostname let hostname_to_check = match net_perm_token { Some(token) => token.check_host(&addr.hostname).to_string(), None => addr.hostname.clone(), }; state_ .borrow_mut::<PermissionsContainer>() .check_net(&(&hostname_to_check, Some(addr.port)), "Deno.connect()")?; } let options = options.unwrap_or_default(); // Fetch the cancel handle before the first await point. Aborting from JS // closes the resource, so fetching it any later would miss an abort that // fires while an earlier future (e.g. DNS resolution) is still pending. let cancel_handle = resource_abort_id.and_then(|rid| { state .borrow_mut() .resource_table .get::<CancelHandle>(rid) .ok() }); let tcp_stream_result = async { // Resolve all addresses for Happy Eyeballs let resolve_fut = resolve_addr(&addr.hostname, addr.port); let addrs: Vec<SocketAddr> = if let Some(cancel_handle) = &cancel_handle { resolve_fut.or_cancel(cancel_handle).await??.collect() } else { resolve_fut.await?.collect() }; if addrs.is_empty() { return Err(NetError::NoResolvedAddress); } // Happy Eyeballs races every resolved candidate, so it may connect to any // of them; the non-racing path only ever connects to `addrs[0]`. let use_happy_eyeballs = options.auto_select_family && addrs.len() > 1; // Post-resolution deny check: verify the IPs we may actually connect to // are not denied. This prevents bypassing IP-literal deny rules via // numeric hostname aliases (e.g. 2130706433 -> 127.0.0.1). Only the // candidates we may attempt are checked: all of them when Happy Eyeballs // races them, otherwise just the single address that will be used. { let mut state_ = state.borrow_mut(); let permissions = state_.borrow_mut::<PermissionsContainer>(); let checked = if use_happy_eyeballs { &addrs[..] } else { &addrs[..1] }; for addr in checked { permissions.check_net_resolved( &addr.ip(), addr.port(), "Deno.connect()", )?; } } // Use Happy Eyeballs if enabled and multiple addresses available if use_happy_eyeballs { let attempt_delay = Duration::from_millis(options.auto_select_family_attempt_delay); crate::happy_eyeballs::connect_happy_eyeballs( addrs, attempt_delay, cancel_handle.clone(), ) .await .map(|result| result.stream) .map_err(NetError::Io) } else { // Single address or Happy Eyeballs disabled - use first address let addr = addrs[0]; if let Some(cancel_handle) = &cancel_handle { match TcpStream::connect(&addr).or_cancel(cancel_handle).await { Ok(result) => result.map_err(NetError::Io), Err(err) => Err(NetError::Canceled(err)), } } else { TcpStream::connect(&addr).await.map_err(NetError::Io) } } } .await; // Remove the cancel handle resource on every exit path (including // resolution errors) so it does not leak in the resource table. if let Some(cancel_rid) = resource_abort_id && let Ok(res) = state.borrow_mut().resource_table.take_any(cancel_rid) { res.close(); } let tcp_stream = tcp_stream_result?; let local_addr = tcp_stream.local_addr()?; let remote_addr = tcp_stream.peer_addr()?; let mut state_ = state.borrow_mut(); let rid = state_ .resource_table .add(TcpStreamResource::new(tcp_stream.into_split())); Ok((rid, IpAddr::from(local_addr), IpAddr::from(remote_addr))) } struct UdpSocketResource { socket: AsyncRefCell<UdpSocket>, cancel: CancelHandle, } impl Resource for UdpSocketResource { fn name(&self) -> Cow<'_, str> { "udpSocket".into() } fn close(self: Rc<Self>) { self.cancel.cancel() } } #[op2(stack_trace)] pub fn op_net_listen_tcp( state: &mut OpState, #[scoped] addr: IpAddr, reuse_port: bool, load_balanced: bool, tcp_backlog: i32, ) -> Result<(ResourceId, IpAddr), NetError> { if reuse_port { super::check_unstable(state, "Deno.listen({ reusePort: true })"); } state .borrow_mut::<PermissionsContainer>() .check_net(&(&addr.hostname, Some(addr.port)), "Deno.listen()")?; let addr = resolve_addr_sync(&addr.hostname, addr.port)? .next() .ok_or_else(|| NetError::NoResolvedAddress)?; state .borrow_mut::<PermissionsContainer>() .check_net_resolved(&addr.ip(), addr.port(), "Deno.listen()")?; let listener = if load_balanced { TcpListener::bind_load_balanced(addr, tcp_backlog) } else { TcpListener::bind_direct(addr, reuse_port, tcp_backlog) }?; let local_addr = listener.local_addr()?; let listener_resource = NetworkListenerResource::new(listener); let rid = state.resource_table.add(listener_resource); Ok((rid, IpAddr::from(local_addr))) } fn net_listen_udp( state: &mut OpState, addr: IpAddr, reuse_address: bool, loopback: bool, ) -> Result<(ResourceId, IpAddr), NetError> { state .borrow_mut::<PermissionsContainer>() .check_net(&(&addr.hostname, Some(addr.port)), "Deno.listenDatagram()")?; let addr = resolve_addr_sync(&addr.hostname, addr.port)? .next() .ok_or_else(|| NetError::NoResolvedAddress)?; state .borrow_mut::<PermissionsContainer>() .check_net_resolved(&addr.ip(), addr.port(), "Deno.listenDatagram()")?; let domain = if addr.is_ipv4() { Domain::IPV4 } else { Domain::IPV6 }; let socket_tmp = Socket::new(domain, Type::DGRAM, Some(Protocol::UDP))?; if reuse_address { // This logic is taken from libuv: // // On the BSDs, SO_REUSEPORT implies SO_REUSEADDR but with some additional // refinements for programs that use multicast. // // Linux as of 3.9 has a SO_REUSEPORT socket option but with semantics that // are different from the BSDs: it _shares_ the port rather than steal it // from the current listener. While useful, it's not something we can // emulate on other platforms so we don't enable it. #[cfg(any( target_os = "windows", target_os = "android", target_os = "linux" ))] socket_tmp.set_reuse_address(true)?; #[cfg(all(unix, not(any(target_os = "android", target_os = "linux"))))] socket_tmp.set_reuse_port(true)?; } let socket_addr = socket2::SockAddr::from(addr); socket_tmp.bind(&socket_addr)?; socket_tmp.set_nonblocking(true)?; // Enable messages to be sent to the broadcast address (255.255.255.255) by default socket_tmp.set_broadcast(true)?; if domain == Domain::IPV4 { socket_tmp.set_multicast_loop_v4(loopback)?; } else { socket_tmp.set_multicast_loop_v6(loopback)?; } let std_socket: std::net::UdpSocket = socket_tmp.into(); let socket = UdpSocket::from_std(std_socket)?; let local_addr = socket.local_addr()?; let socket_resource = UdpSocketResource { socket: AsyncRefCell::new(socket), cancel: Default::default(), }; let rid = state.resource_table.add(socket_resource); Ok((rid, IpAddr::from(local_addr))) } #[op2(stack_trace)] pub fn op_net_listen_udp( state: &mut OpState, #[scoped] addr: IpAddr, reuse_address: bool, loopback: bool, ) -> Result<(ResourceId, IpAddr), NetError> { super::check_unstable(state, "Deno.listenDatagram"); net_listen_udp(state, addr, reuse_address, loopback) } #[op2(stack_trace)] pub fn op_node_unstable_net_listen_udp( state: &mut OpState, #[scoped] addr: IpAddr, reuse_address: bool, loopback: bool, ) -> Result<(ResourceId, IpAddr), NetError> { net_listen_udp(state, addr, reuse_address, loopback) } #[cfg(any(target_os = "android", target_os = "linux", target_os = "macos"))] #[op2(stack_trace)] pub async fn op_net_connect_vsock( state: Rc<RefCell<OpState>>, #[smi] cid: u32, #[smi] port: u32, ) -> Result<(ResourceId, (u32, u32), (u32, u32)), NetError> { use std::sync::Arc; use deno_features::FeatureChecker; use tokio_vsock::VsockAddr; use tokio_vsock::VsockStream; state .borrow() .borrow::<Arc<FeatureChecker>>() .check_or_exit("vsock", "Deno.connect"); state .borrow_mut() .borrow_mut::<PermissionsContainer>() .check_net_vsock(cid, port, "Deno.connect()")?; let addr = VsockAddr::new(cid, port); let vsock_stream = VsockStream::connect(addr).await?; let local_addr = vsock_stream.local_addr()?; let remote_addr = vsock_stream.peer_addr()?; let rid = state .borrow_mut() .resource_table .add(crate::io::VsockStreamResource::new( vsock_stream.into_split(), )); Ok(( rid, (local_addr.cid(), local_addr.port()), (remote_addr.cid(), remote_addr.port()), )) } #[cfg(not(any( target_os = "android", target_os = "linux", target_os = "macos" )))] #[op2(fast)] pub fn op_net_connect_vsock() -> Result<(), NetError> { Err(NetError::VsockUnsupported) } #[cfg(any(target_os = "android", target_os = "linux", target_os = "macos"))] #[op2(stack_trace)] pub fn op_net_listen_vsock( state: &mut OpState, #[smi] cid: u32, #[smi] port: u32, ) -> Result<(ResourceId, u32, u32), NetError> { use std::sync::Arc; use deno_features::FeatureChecker; use tokio_vsock::VsockAddr; use tokio_vsock::VsockListener; state .borrow::<Arc<FeatureChecker>>() .check_or_exit("vsock", "Deno.listen"); state.borrow_mut::<PermissionsContainer>().check_net_vsock( cid, port, "Deno.listen()", )?; let addr = VsockAddr::new(cid, port); let listener = VsockListener::bind(addr)?; let local_addr = listener.local_addr()?; let listener_resource = NetworkListenerResource::new(listener); let rid = state.resource_table.add(listener_resource); Ok((rid, local_addr.cid(), local_addr.port())) } #[cfg(not(any( target_os = "android", target_os = "linux", target_os = "macos" )))] #[op2(fast)] pub fn op_net_listen_vsock() -> Result<(), NetError> { Err(NetError::VsockUnsupported) } #[cfg(any(target_os = "android", target_os = "linux", target_os = "macos"))] #[op2] pub async fn op_net_accept_vsock( state: Rc<RefCell<OpState>>, #[smi] rid: ResourceId, ) -> Result<(ResourceId, (u32, u32), (u32, u32)), NetError> { use tokio_vsock::VsockListener; let resource = state .borrow() .resource_table .get::<NetworkListenerResource<VsockListener>>(rid) .map_err(|_| NetError::ListenerClosed)?; let listener = RcRef::map(&resource, |r| &r.listener) .try_borrow_mut() .ok_or_else(|| NetError::AcceptTaskOngoing)?; let cancel = RcRef::map(resource, |r| &r.cancel); let (vsock_stream, _socket_addr) = listener .accept() .try_or_cancel(cancel) .await .map_err(accept_err)?; let local_addr = vsock_stream.local_addr()?; let remote_addr = vsock_stream.peer_addr()?; let mut state = state.borrow_mut(); let rid = state .resource_table .add(crate::io::VsockStreamResource::new( vsock_stream.into_split(), )); Ok(( rid, (local_addr.cid(), local_addr.port()), (remote_addr.cid(), remote_addr.port()), )) } #[cfg(not(any( target_os = "android", target_os = "linux", target_os = "macos" )))] #[op2(fast)] pub fn op_net_accept_vsock() -> Result<(), NetError> { Err(NetError::VsockUnsupported) } #[op2] pub fn op_net_listen_tunnel( state: &mut OpState, ) -> Result<(ResourceId, IpAddr), NetError> { let Some(listener) = super::tunnel::get_tunnel() else { return Err(NetError::TunnelMissing); }; let listener = listener.clone(); let local_addr = listener.local_addr()?.into(); let rid = state .resource_table .add(crate::raw::NetworkListenerResource::new(listener)); Ok((rid, local_addr)) } #[op2] pub async fn op_net_accept_tunnel( state: Rc<RefCell<OpState>>, #[smi] rid: ResourceId, ) -> Result<(ResourceId, IpAddr, IpAddr), NetError> { let resource = state .borrow() .resource_table .get::<NetworkListenerResource<crate::tunnel::TunnelConnection>>(rid) .map_err(|_| NetError::ListenerClosed)?; let listener = RcRef::map(&resource, |r| &r.listener) .try_borrow_mut() .ok_or_else(|| NetError::AcceptTaskOngoing)?; let cancel = RcRef::map(resource, |r| &r.cancel); let (stream, _) = listener .accept() .try_or_cancel(cancel) .await .map_err(accept_err)?; let local_addr = stream.local_addr()?; let remote_addr = stream.peer_addr()?; let rid = state .borrow_mut() .resource_table .add(crate::tunnel::TunnelStreamResource::new(stream)); Ok((rid, local_addr.into(), remote_addr.into())) } #[derive(ToV8, Eq, PartialEq, Debug)] #[to_v8(untagged)] pub enum DnsRecordData { A(String), Aaaa(String), Aname(String), Caa { critical: bool, tag: String, value: String, }, Cname(String), Mx { preference: u16, exchange: String, }, Naptr { order: u16, preference: u16, flags: String, services: String, regexp: String, replacement: String, }, Ns(String), Ptr(String), Soa { mname: String, rname: String, serial: u32, refresh: i32, retry: i32, expire: i32, minimum: u32, }, Srv { priority: u16, weight: u16, port: u16, target: String, }, Txt(Vec<String>), } #[derive(ToV8, Eq, PartialEq, Debug)] pub struct DnsRecordWithTtl { pub data: DnsRecordData, /// Record type name, populated for ANY queries to distinguish /// untagged string variants (A vs AAAA vs NS vs PTR vs CNAME). #[to_v8(serde)] pub record_type: Option<String>, pub ttl: u32, } #[derive(FromV8)] pub struct ResolveAddrArgs { cancel_rid: Option<ResourceId>, query: String, #[from_v8(serde)] record_type: RecordType, options: Option<ResolveDnsOption>, } #[derive(FromV8)] pub struct ResolveDnsOption { name_server: Option<NameServer>, } #[derive(FromV8)] pub struct NameServer { ip_addr: String, #[from_v8(default = 53)] port: u16, } #[op2(stack_trace)] pub async fn op_dns_resolve( state: Rc<RefCell<OpState>>, #[scoped] args: ResolveAddrArgs, use_edns: bool, ) -> Result<Vec<DnsRecordWithTtl>, NetError> { let ResolveAddrArgs { query, record_type, options, cancel_rid, } = args; let (config, mut opts) = if let Some(name_server) = options.as_ref().and_then(|o| o.name_server.as_ref()) { let group = NameServerConfigGroup::from_ips_clear( &[name_server.ip_addr.parse()?], name_server.port, true, ); (ResolverConfig::from_parts(None, vec![], group), { let mut opts = ResolverOpts::default(); if use_edns { opts.edns0 = true; } opts }) } else { system_conf::read_system_conf()? }; // When a cancel handle is provided, use a short resolver timeout so // that hickory's background connection tasks clean up quickly after // cancellation (they are not aborted by the cancel handle itself). if cancel_rid.is_some() { opts.timeout = std::time::Duration::from_secs(1); opts.attempts = 1; } { let mut s = state.borrow_mut(); let perm = s.borrow_mut::<PermissionsContainer>(); // Checks permission against the name servers which will be actually queried. for ns in config.name_servers() { let socker_addr = &ns.socket_addr; let ip = socker_addr.ip().to_string(); let port = socker_addr.port(); perm.check_net(&(&ip, Some(port)), "Deno.resolveDns()")?; } } let provider = TokioConnectionProvider::default(); let resolver = hickory_resolver::Resolver::builder_with_config(config, provider) .with_options(opts) .build(); let lookup_fut = resolver.lookup(query, record_type); let cancel_handle = cancel_rid.and_then(|rid| { state .borrow_mut() .resource_table .get::<CancelHandle>(rid) .ok() }); let lookup = if let Some(cancel_handle) = cancel_handle { let lookup_rv = lookup_fut.or_cancel(cancel_handle).await; if let Some(cancel_rid) = cancel_rid && let Ok(res) = state.borrow_mut().resource_table.take_any(cancel_rid) { res.close(); }; lookup_rv? } else { lookup_fut.await }; let lookup = lookup.map_err(|e| match e.kind() { ResolveErrorKind::Proto(ProtoError { kind, .. }) if matches!(**kind, ProtoErrorKind::NoRecordsFound { .. }) => { NetError::DnsNotFound(e) } ResolveErrorKind::Proto(ProtoError { kind, .. }) if matches!(**kind, ProtoErrorKind::NoConnections) => { NetError::DnsNotConnected(e) } ResolveErrorKind::Proto(ProtoError { kind, .. }) if matches!(**kind, ProtoErrorKind::Timeout) => { NetError::DnsTimedOut(e) } _ => NetError::Dns(e), })?; format_dns_records(lookup.records(), record_type) } fn format_dns_records<'a>( records: impl IntoIterator<Item = &'a Record>, record_type: RecordType, ) -> Result<Vec<DnsRecordWithTtl>, NetError> { records .into_iter() .filter_map(|rec| { let is_any = record_type == RecordType::ANY; // For ANY queries, use each record's actual type for formatting let effective_type = if is_any { rec.record_type() } else { record_type }; let r = format_rdata(effective_type)(rec.data()).transpose(); r.map(|maybe_data| { maybe_data.map(|data| DnsRecordWithTtl { data, record_type: if is_any { Some(effective_type.to_string()) } else { None }, ttl: rec.ttl(), }) }) }) .collect::<Result<Vec<DnsRecordWithTtl>, NetError>>() } #[op2] #[serde] pub fn op_net_get_system_dns_servers( state: &mut OpState, ) -> Result<Vec<(String, u16)>, NetError> { // Reading the host resolver configuration (the system nameservers, e.g. // from /etc/resolv.conf) exposes host network configuration, so gate it // behind the same `sys` permission as Deno.networkInterfaces(). Reachable // from JS via node:dns.getServers(). state .borrow_mut::<PermissionsContainer>() .check_sys("networkInterfaces", "node:dns.getServers()")?; let (config, _opts) = system_conf::read_system_conf() .unwrap_or_else(|_| (ResolverConfig::default(), ResolverOpts::default())); let servers = config .name_servers() .iter() .map(|ns| { let addr = ns.socket_addr; (addr.ip().to_string(), addr.port()) }) .collect(); Ok(servers) } #[op2(fast)] pub fn op_set_nodelay( state: &mut OpState, #[smi] rid: ResourceId, nodelay: bool, ) -> Result<(), NetError> { op_set_nodelay_inner(state, rid, nodelay) } #[inline] pub fn op_set_nodelay_inner( state: &mut OpState, rid: ResourceId, nodelay: bool, ) -> Result<(), NetError> { let resource: Rc<TcpStreamResource> = state.resource_table.get::<TcpStreamResource>(rid)?; resource.set_nodelay(nodelay).map_err(NetError::Map) } #[op2(fast)] pub fn op_set_keepalive( state: &mut OpState, #[smi] rid: ResourceId, keepalive: bool, ) -> Result<(), NetError> { op_set_keepalive_inner(state, rid, keepalive) } #[inline] pub fn op_set_keepalive_inner( state: &mut OpState, rid: ResourceId, keepalive: bool, ) -> Result<(), NetError> { let resource: Rc<TcpStreamResource> = state.resource_table.get::<TcpStreamResource>(rid)?; resource.set_keepalive(keepalive).map_err(NetError::Map) } fn format_rdata( ty: RecordType, ) -> impl Fn(&RData) -> Result<Option<DnsRecordData>, NetError> { use RecordType::*; move |r: &RData| -> Result<Option<DnsRecordData>, NetError> { let record = match ty { A => r.as_a().map(ToString::to_string).map(DnsRecordData::A), AAAA => r .as_aaaa() .map(ToString::to_string) .map(DnsRecordData::Aaaa), ANAME => r .as_aname() .map(ToString::to_string) .map(DnsRecordData::Aname), CAA => r.as_caa().map(|caa| DnsRecordData::Caa { critical: caa.issuer_critical(), tag: caa.tag().to_string(), value: String::from_utf8_lossy(caa.raw_value()).into_owned(), }), CNAME => r .as_cname() .map(ToString::to_string) .map(DnsRecordData::Cname), MX => r.as_mx().map(|mx| DnsRecordData::Mx { preference: mx.preference(), exchange: mx.exchange().to_string(), }), NAPTR => r.as_naptr().map(|naptr| DnsRecordData::Naptr { order: naptr.order(), preference: naptr.preference(), flags: String::from_utf8_lossy(naptr.flags()).into_owned(), services: String::from_utf8_lossy(naptr.services()).into_owned(), regexp: String::from_utf8_lossy(naptr.regexp()).into_owned(), replacement: naptr.replacement().to_string(), }), NS => r.as_ns().map(ToString::to_string).map(DnsRecordData::Ns), PTR => r.as_ptr().map(ToString::to_string).map(DnsRecordData::Ptr), SOA => r.as_soa().map(|soa| DnsRecordData::Soa { mname: soa.mname().to_string(), rname: soa.rname().to_string(), serial: soa.serial(), refresh: soa.refresh(), retry: soa.retry(), expire: soa.expire(), minimum: soa.minimum(), }), SRV => r.as_srv().map(|srv| DnsRecordData::Srv { priority: srv.priority(), weight: srv.weight(), port: srv.port(), target: srv.target().to_string(), }), TXT => r.as_txt().map(|txt| { let texts: Vec<String> = txt .iter() .map(|bytes| { // Tries to parse these bytes as Latin-1 bytes.iter().map(|&b| b as char).collect::<String>() }) .collect(); DnsRecordData::Txt(texts) }), _ => return Err(NetError::UnsupportedRecordType), }; Ok(record) } } #[cfg(test)] mod tests { use std::net::Ipv4Addr; use std::net::Ipv6Addr; use std::net::ToSocketAddrs; use std::sync::Mutex; use deno_core::JsRuntime; use deno_core::RuntimeOptions; use deno_core::futures::FutureExt; use hickory_proto::rr::Name; use hickory_proto::rr::rdata::SOA; use hickory_proto::rr::rdata::a::A; use hickory_proto::rr::rdata::aaaa::AAAA; use hickory_proto::rr::rdata::caa::CAA; use hickory_proto::rr::rdata::caa::KeyValue; use hickory_proto::rr::rdata::mx::MX; use hickory_proto::rr::rdata::name::ANAME; use hickory_proto::rr::rdata::name::CNAME; use hickory_proto::rr::rdata::name::NS; use hickory_proto::rr::rdata::name::PTR; use hickory_proto::rr::rdata::naptr::NAPTR; use hickory_proto::rr::rdata::srv::SRV; use hickory_proto::rr::rdata::txt::TXT; use hickory_proto::rr::record_data::RData; use hickory_proto::serialize::binary::BinDecoder; use hickory_proto::serialize::binary::Restrict; use socket2::SockRef; use super::*; #[test] fn rdata_to_return_record_a() { let func = format_rdata(RecordType::A); let rdata = RData::A(A(Ipv4Addr::new(127, 0, 0, 1))); assert_eq!( func(&rdata).unwrap(), Some(DnsRecordData::A("127.0.0.1".to_string())) ); } #[test] fn rdata_to_return_record_aaaa() { let func = format_rdata(RecordType::AAAA); let rdata = RData::AAAA(AAAA(Ipv6Addr::new(0, 0, 0, 0, 0, 0, 0, 1))); assert_eq!( func(&rdata).unwrap(), Some(DnsRecordData::Aaaa("::1".to_string())) ); } #[test] fn rdata_to_return_record_aname() { let func = format_rdata(RecordType::ANAME); let rdata = RData::ANAME(ANAME(Name::new())); assert_eq!( func(&rdata).unwrap(), Some(DnsRecordData::Aname("".to_string())) ); } #[test] fn rdata_to_return_record_caa() { let func = format_rdata(RecordType::CAA); let rdata = RData::CAA(CAA::new_issue( false, Some(Name::parse("example.com", None).unwrap()), vec![KeyValue::new("account", "123456")], )); assert_eq!( func(&rdata).unwrap(), Some(DnsRecordData::Caa { critical: false, tag: "issue".to_string(), value: "example.com; account=123456".to_string(), }) ); } #[test] fn dns_records_caa_lossy_invalid_utf8_for_any_query() { // Raw CAA RDATA received from the network: flags, tag, then an opaque // value containing invalid UTF-8. const RDATA: &[u8] = b"\x00\x05issue\xffa"; let rdata = RData::read( &mut BinDecoder::new(RDATA), RecordType::CAA, Restrict::new(u16::try_from(RDATA.len()).unwrap()), ) .unwrap(); let records = [Record::from_rdata(Name::new(), 60, rdata)]; assert_eq!( format_dns_records(&records, RecordType::ANY).unwrap(), vec![DnsRecordWithTtl { data: DnsRecordData::Caa { critical: false, tag: "issue".to_string(), value: "\u{fffd}a".to_string(), }, record_type: Some("CAA".to_string()), ttl: 60, }] ); } #[test] fn rdata_to_return_record_cname() { let func = format_rdata(RecordType::CNAME); let rdata = RData::CNAME(CNAME(Name::new())); assert_eq!( func(&rdata).unwrap(), Some(DnsRecordData::Cname("".to_string())) ); } #[test] fn rdata_to_return_record_mx() { let func = format_rdata(RecordType::MX); let rdata = RData::MX(MX::new(10, Name::new())); assert_eq!( func(&rdata).unwrap(), Some(DnsRecordData::Mx { preference: 10, exchange: "".to_string() }) ); } #[test] fn rdata_to_return_record_naptr() { let func = format_rdata(RecordType::NAPTR); let rdata = RData::NAPTR(NAPTR::new( 1, 2, <Box<[u8]>>::default(), <Box<[u8]>>::default(), <Box<[u8]>>::default(), Name::new(), )); assert_eq!( func(&rdata).unwrap(), Some(DnsRecordData::Naptr { order: 1, preference: 2, flags: "".to_string(), services: "".to_string(), regexp: "".to_string(), replacement: "".to_string() }) ); } #[test] fn rdata_to_return_record_naptr_lossy_invalid_utf8() { // Flags are restricted to ASCII by Hickory's wire decoder. Services and // regexp remain arbitrary character strings and can contain invalid UTF-8. const RDATA: &[u8] = b"\x00\x01\x00\x02\x01U\x02s\xff\x02\xffr\x00"; let rdata = RData::read( &mut BinDecoder::new(RDATA), RecordType::NAPTR, Restrict::new(u16::try_from(RDATA.len()).unwrap()), ) .unwrap(); assert_eq!( format_rdata(RecordType::NAPTR)(&rdata).unwrap(), Some(DnsRecordData::Naptr { order: 1, preference: 2, flags: "U".to_string(), services: "s\u{fffd}".to_string(), regexp: "\u{fffd}r".to_string(), replacement: ".".to_string(), }) ); } #[test] fn rdata_to_return_record_ns() { let func = format_rdata(RecordType::NS); let rdata = RData::NS(NS(Name::new())); assert_eq!( func(&rdata).unwrap(), Some(DnsRecordData::Ns("".to_string())) ); } #[test] fn rdata_to_return_record_ptr() { let func = format_rdata(RecordType::PTR); let rdata = RData::PTR(PTR(Name::new())); assert_eq!( func(&rdata).unwrap(), Some(DnsRecordData::Ptr("".to_string())) ); } #[test] fn rdata_to_return_record_soa() { let func = format_rdata(RecordType::SOA); let rdata = RData::SOA(SOA::new( Name::new(), Name::new(), 0, i32::MAX, i32::MAX, i32::MAX, 0, )); assert_eq!( func(&rdata).unwrap(), Some(DnsRecordData::Soa { mname: "".to_string(), rname: "".to_string(), serial: 0, refresh: i32::MAX, retry: i32::MAX, expire: i32::MAX, minimum: 0, }) ); } #[test] fn rdata_to_return_record_srv() { let func = format_rdata(RecordType::SRV); let rdata = RData::SRV(SRV::new(1, 2, 3, Name::new())); assert_eq!( func(&rdata).unwrap(), Some(DnsRecordData::Srv { priority: 1, weight: 2, port: 3, target: "".to_string() }) ); } #[test] fn rdata_to_return_record_txt() { let func = format_rdata(RecordType::TXT); let rdata = RData::TXT(TXT::from_bytes(vec![ "foo".as_bytes(), "bar".as_bytes(), &[0xa3], // "£" in Latin-1 &[0xe3, 0x81, 0x82], // "あ" in UTF-8 ])); assert_eq!( func(&rdata).unwrap(), Some(DnsRecordData::Txt(vec![ "foo".to_string(), "bar".to_string(), "£".to_string(), "ã\u{81}\u{82}".to_string(), ])) ); } #[tokio::test(flavor = "multi_thread", worker_threads = 1)] async fn tcp_set_no_delay() { let set_nodelay = Box::new(|state: &mut OpState, rid| { op_set_nodelay_inner(state, rid, true).unwrap(); }); let test_fn = Box::new(|socket: SockRef| { assert!(socket.nodelay().unwrap()); assert!(!socket.keepalive().unwrap()); }); check_sockopt(String::from("127.0.0.1:4145"), set_nodelay, test_fn).await; } #[tokio::test(flavor = "multi_thread", worker_threads = 1)] async fn tcp_set_keepalive() { let set_keepalive = Box::new(|state: &mut OpState, rid| { op_set_keepalive_inner(state, rid, true).unwrap(); }); let test_fn = Box::new(|socket: SockRef| { assert!(!socket.nodelay().unwrap()); assert!(socket.keepalive().unwrap()); }); check_sockopt(String::from("127.0.0.1:4146"), set_keepalive, test_fn).await; } #[allow( clippy::type_complexity, reason = "set_sockopt_fn isn't that complex" )] async fn check_sockopt( addr: String, set_sockopt_fn: Box<dyn Fn(&mut OpState, u32)>, test_fn: Box<dyn FnOnce(SockRef)>, ) { let sockets = Arc::new(Mutex::new(vec![])); let clone_addr = addr.clone(); let addr = addr.to_socket_addrs().unwrap().next().unwrap(); let listener = TcpListener::bind_direct(addr, false, 511).unwrap(); let accept_fut = listener.accept().boxed_local(); let store_fut = async move { let socket = accept_fut.await.unwrap(); sockets.lock().unwrap().push(socket); } .boxed_local(); use std::sync::Arc; use deno_permissions::RuntimePermissionDescriptorParser; deno_core::extension!( test_ext, state = |state| { let parser = Arc::new(RuntimePermissionDescriptorParser::new( sys_traits::impls::RealSys, )); state.put(PermissionsContainer::allow_all(parser)); } ); let runtime = JsRuntime::new(RuntimeOptions { extensions: vec![test_ext::init()], ..Default::default() }); let conn_state = runtime.op_state(); let server_addr: Vec<&str> = clone_addr.split(':').collect(); let ip_addr = IpAddr { hostname: String::from(server_addr[0]), port: server_addr[1].parse().unwrap(), }; let mut connect_fut = op_net_connect_tcp_inner(conn_state, ip_addr, None, None, None) .boxed_local(); let mut rid = None; tokio::select! { _ = store_fut => { let result = connect_fut.await; let vals = result.unwrap(); rid = rid.or(Some(vals.0)); }, result = &mut connect_fut => { let vals = result.unwrap(); rid = rid.or(Some(vals.0)); } } let rid = rid.unwrap(); let state = runtime.op_state(); set_sockopt_fn(&mut state.borrow_mut(), rid); let resource = state .borrow_mut() .resource_table .get::<TcpStreamResource>(rid) .unwrap(); let wr = resource.wr_borrow_mut().await; let stream = wr.as_ref().as_ref(); let socket = socket2::SockRef::from(stream); test_fn(socket); } #[tokio::test(flavor = "multi_thread", worker_threads = 1)] async fn tcp_connect_cancel_while_resolving() { use std::sync::Arc; use std::task::Poll; use deno_permissions::RuntimePermissionDescriptorParser; deno_core::extension!( test_ext, state = |state| { let parser = Arc::new(RuntimePermissionDescriptorParser::new( sys_traits::impls::RealSys, )); state.put(PermissionsContainer::allow_all(parser)); } ); let runtime = JsRuntime::new(RuntimeOptions { extensions: vec![test_ext::init()], ..Default::default() }); let state = runtime.op_state(); let cancel_rid = state.borrow_mut().resource_table.add(CancelHandle::new()); // A hostname (not an IP literal) forces async resolution, so the first // poll parks the op in the DNS lookup. let ip_addr = IpAddr { hostname: String::from("localhost"), port: 1, }; let mut connect_fut = op_net_connect_tcp_inner( state.clone(), ip_addr, None, Some(cancel_rid), None, ) .boxed_local(); let pending = std::future::poll_fn(|cx| { Poll::Ready(connect_fut.poll_unpin(cx).is_pending()) }) .await; assert!(pending); // Abort the same way the JS abort handler does: close the cancel handle // resource while the op is still resolving the address. let res = state .borrow_mut() .resource_table .take_any(cancel_rid) .unwrap(); res.close(); let result = connect_fut.await; assert!(matches!(result, Err(NetError::Canceled(_)))); } }