blob: 442f495171b4ae4ba2c1cc87c31c448b16a57928 [file] [log] [blame]
use std::convert::TryInto;
use std::io;
use std::mem::size_of;
use std::net::{self, Ipv4Addr, Ipv6Addr, SocketAddr, SocketAddrV4, SocketAddrV6};
use std::os::windows::io::FromRawSocket;
use std::os::windows::raw::SOCKET as StdSocket;
use std::ptr;
use std::time::Duration; // winapi uses usize, stdlib uses u32/u64.
use winapi::ctypes::{c_char, c_int, c_ulong, c_ushort};
use winapi::shared::mstcpip;
use winapi::shared::ws2def::{AF_INET, AF_INET6, SOCKADDR_IN, SOCKADDR_STORAGE};
use winapi::shared::ws2ipdef::SOCKADDR_IN6_LH;
use winapi::shared::minwindef::{BOOL, DWORD, FALSE, LPDWORD, LPVOID, TRUE};
use winapi::um::winsock2::{
self, closesocket, getsockname, getsockopt, linger, setsockopt, WSAIoctl, LPWSAOVERLAPPED,
PF_INET, PF_INET6, SOCKET, SOCKET_ERROR, SOCK_STREAM, SOL_SOCKET, SO_KEEPALIVE, SO_LINGER,
SO_RCVBUF, SO_REUSEADDR, SO_SNDBUF,
};
use crate::net::TcpKeepalive;
use crate::sys::windows::net::{init, new_socket, socket_addr};
pub(crate) type TcpSocket = SOCKET;
pub(crate) fn new_v4_socket() -> io::Result<TcpSocket> {
init();
new_socket(PF_INET, SOCK_STREAM)
}
pub(crate) fn new_v6_socket() -> io::Result<TcpSocket> {
init();
new_socket(PF_INET6, SOCK_STREAM)
}
pub(crate) fn bind(socket: TcpSocket, addr: SocketAddr) -> io::Result<()> {
use winsock2::bind;
let (raw_addr, raw_addr_length) = socket_addr(&addr);
syscall!(
bind(socket, raw_addr.as_ptr(), raw_addr_length),
PartialEq::eq,
SOCKET_ERROR
)?;
Ok(())
}
pub(crate) fn connect(socket: TcpSocket, addr: SocketAddr) -> io::Result<net::TcpStream> {
use winsock2::connect;
let (raw_addr, raw_addr_length) = socket_addr(&addr);
let res = syscall!(
connect(socket, raw_addr.as_ptr(), raw_addr_length),
PartialEq::eq,
SOCKET_ERROR
);
match res {
Err(err) if err.kind() != io::ErrorKind::WouldBlock => Err(err),
_ => Ok(unsafe { net::TcpStream::from_raw_socket(socket as StdSocket) }),
}
}
pub(crate) fn listen(socket: TcpSocket, backlog: u32) -> io::Result<net::TcpListener> {
use std::convert::TryInto;
use winsock2::listen;
let backlog = backlog.try_into().unwrap_or(i32::max_value());
syscall!(listen(socket, backlog), PartialEq::eq, SOCKET_ERROR)?;
Ok(unsafe { net::TcpListener::from_raw_socket(socket as StdSocket) })
}
pub(crate) fn close(socket: TcpSocket) {
let _ = unsafe { closesocket(socket) };
}
pub(crate) fn set_reuseaddr(socket: TcpSocket, reuseaddr: bool) -> io::Result<()> {
let val: BOOL = if reuseaddr { TRUE } else { FALSE };
match unsafe {
setsockopt(
socket,
SOL_SOCKET,
SO_REUSEADDR,
&val as *const _ as *const c_char,
size_of::<BOOL>() as c_int,
)
} {
SOCKET_ERROR => Err(io::Error::last_os_error()),
_ => Ok(()),
}
}
pub(crate) fn get_reuseaddr(socket: TcpSocket) -> io::Result<bool> {
let mut optval: c_char = 0;
let mut optlen = size_of::<BOOL>() as c_int;
match unsafe {
getsockopt(
socket,
SOL_SOCKET,
SO_REUSEADDR,
&mut optval as *mut _ as *mut _,
&mut optlen,
)
} {
SOCKET_ERROR => Err(io::Error::last_os_error()),
_ => Ok(optval != 0),
}
}
pub(crate) fn get_localaddr(socket: TcpSocket) -> io::Result<SocketAddr> {
let mut storage: SOCKADDR_STORAGE = unsafe { std::mem::zeroed() };
let mut length = std::mem::size_of_val(&storage) as c_int;
match unsafe { getsockname(socket, &mut storage as *mut _ as *mut _, &mut length) } {
SOCKET_ERROR => Err(io::Error::last_os_error()),
_ => {
if storage.ss_family as c_int == AF_INET {
// Safety: if the ss_family field is AF_INET then storage must be a sockaddr_in.
let addr: &SOCKADDR_IN = unsafe { &*(&storage as *const _ as *const SOCKADDR_IN) };
let ip_bytes = unsafe { addr.sin_addr.S_un.S_un_b() };
let ip =
Ipv4Addr::from([ip_bytes.s_b1, ip_bytes.s_b2, ip_bytes.s_b3, ip_bytes.s_b4]);
let port = u16::from_be(addr.sin_port);
Ok(SocketAddr::V4(SocketAddrV4::new(ip, port)))
} else if storage.ss_family as c_int == AF_INET6 {
// Safety: if the ss_family field is AF_INET6 then storage must be a sockaddr_in6.
let addr: &SOCKADDR_IN6_LH =
unsafe { &*(&storage as *const _ as *const SOCKADDR_IN6_LH) };
let ip = Ipv6Addr::from(*unsafe { addr.sin6_addr.u.Byte() });
let port = u16::from_be(addr.sin6_port);
let scope_id = unsafe { *addr.u.sin6_scope_id() };
Ok(SocketAddr::V6(SocketAddrV6::new(
ip,
port,
addr.sin6_flowinfo,
scope_id,
)))
} else {
Err(std::io::ErrorKind::InvalidInput.into())
}
}
}
}
pub(crate) fn set_linger(socket: TcpSocket, dur: Option<Duration>) -> io::Result<()> {
let val: linger = linger {
l_onoff: if dur.is_some() { 1 } else { 0 },
l_linger: dur.map(|dur| dur.as_secs() as c_ushort).unwrap_or_default(),
};
match unsafe {
setsockopt(
socket,
SOL_SOCKET,
SO_LINGER,
&val as *const _ as *const c_char,
size_of::<linger>() as c_int,
)
} {
SOCKET_ERROR => Err(io::Error::last_os_error()),
_ => Ok(()),
}
}
pub(crate) fn get_linger(socket: TcpSocket) -> io::Result<Option<Duration>> {
let mut val: linger = unsafe { std::mem::zeroed() };
let mut len = size_of::<linger>() as c_int;
match unsafe {
getsockopt(
socket,
SOL_SOCKET,
SO_LINGER,
&mut val as *mut _ as *mut _,
&mut len,
)
} {
SOCKET_ERROR => Err(io::Error::last_os_error()),
_ => {
if val.l_onoff == 0 {
Ok(None)
} else {
Ok(Some(Duration::from_secs(val.l_linger as u64)))
}
}
}
}
pub(crate) fn set_recv_buffer_size(socket: TcpSocket, size: u32) -> io::Result<()> {
let size = size.try_into().ok().unwrap_or_else(i32::max_value);
match unsafe {
setsockopt(
socket,
SOL_SOCKET,
SO_RCVBUF,
&size as *const _ as *const c_char,
size_of::<c_int>() as c_int,
)
} {
SOCKET_ERROR => Err(io::Error::last_os_error()),
_ => Ok(()),
}
}
pub(crate) fn get_recv_buffer_size(socket: TcpSocket) -> io::Result<u32> {
let mut optval: c_int = 0;
let mut optlen = size_of::<c_int>() as c_int;
match unsafe {
getsockopt(
socket,
SOL_SOCKET,
SO_RCVBUF,
&mut optval as *mut _ as *mut _,
&mut optlen as *mut _,
)
} {
SOCKET_ERROR => Err(io::Error::last_os_error()),
_ => Ok(optval as u32),
}
}
pub(crate) fn set_send_buffer_size(socket: TcpSocket, size: u32) -> io::Result<()> {
let size = size.try_into().ok().unwrap_or_else(i32::max_value);
match unsafe {
setsockopt(
socket,
SOL_SOCKET,
SO_SNDBUF,
&size as *const _ as *const c_char,
size_of::<c_int>() as c_int,
)
} {
SOCKET_ERROR => Err(io::Error::last_os_error()),
_ => Ok(()),
}
}
pub(crate) fn get_send_buffer_size(socket: TcpSocket) -> io::Result<u32> {
let mut optval: c_int = 0;
let mut optlen = size_of::<c_int>() as c_int;
match unsafe {
getsockopt(
socket,
SOL_SOCKET,
SO_SNDBUF,
&mut optval as *mut _ as *mut _,
&mut optlen as *mut _,
)
} {
SOCKET_ERROR => Err(io::Error::last_os_error()),
_ => Ok(optval as u32),
}
}
pub(crate) fn set_keepalive(socket: TcpSocket, keepalive: bool) -> io::Result<()> {
let val: BOOL = if keepalive { TRUE } else { FALSE };
match unsafe {
setsockopt(
socket,
SOL_SOCKET,
SO_KEEPALIVE,
&val as *const _ as *const c_char,
size_of::<BOOL>() as c_int,
)
} {
SOCKET_ERROR => Err(io::Error::last_os_error()),
_ => Ok(()),
}
}
pub(crate) fn get_keepalive(socket: TcpSocket) -> io::Result<bool> {
let mut optval: c_char = 0;
let mut optlen = size_of::<BOOL>() as c_int;
match unsafe {
getsockopt(
socket,
SOL_SOCKET,
SO_KEEPALIVE,
&mut optval as *mut _ as *mut _,
&mut optlen,
)
} {
SOCKET_ERROR => Err(io::Error::last_os_error()),
_ => Ok(optval != FALSE as c_char),
}
}
pub(crate) fn set_keepalive_params(socket: TcpSocket, keepalive: TcpKeepalive) -> io::Result<()> {
/// Windows configures keepalive time/interval in a u32 of milliseconds.
fn dur_to_ulong_ms(dur: Duration) -> c_ulong {
dur.as_millis()
.try_into()
.ok()
.unwrap_or_else(u32::max_value)
}
// If any of the fields on the `tcp_keepalive` struct were not provided by
// the user, just leaving them zero will clobber any existing value.
// Unfortunately, we can't access the current value, so we will use the
// defaults if a value for the time or interval was not not provided.
let time = keepalive.time.unwrap_or_else(|| {
// The default value is two hours, as per
// https://docs.microsoft.com/en-us/windows/win32/winsock/sio-keepalive-vals
let two_hours = 2 * 60 * 60;
Duration::from_secs(two_hours)
});
let interval = keepalive.interval.unwrap_or_else(|| {
// The default value is one second, as per
// https://docs.microsoft.com/en-us/windows/win32/winsock/sio-keepalive-vals
Duration::from_secs(1)
});
let mut keepalive = mstcpip::tcp_keepalive {
// Enable keepalive
onoff: 1,
keepalivetime: dur_to_ulong_ms(time),
keepaliveinterval: dur_to_ulong_ms(interval),
};
let mut out = 0;
match unsafe {
WSAIoctl(
socket,
mstcpip::SIO_KEEPALIVE_VALS,
&mut keepalive as *mut _ as LPVOID,
size_of::<mstcpip::tcp_keepalive>() as DWORD,
ptr::null_mut() as LPVOID,
0 as DWORD,
&mut out as *mut _ as LPDWORD,
0 as LPWSAOVERLAPPED,
None,
)
} {
0 => Ok(()),
_ => Err(io::Error::last_os_error()),
}
}
pub(crate) fn accept(listener: &net::TcpListener) -> io::Result<(net::TcpStream, SocketAddr)> {
// The non-blocking state of `listener` is inherited. See
// https://docs.microsoft.com/en-us/windows/win32/api/winsock2/nf-winsock2-accept#remarks.
listener.accept()
}