Auto merge of #31945 - sfackler:net2, r=alexcrichton
I have these tagged as stable in 1.9, so this shouldn't merge until the 1.8 beta's cut.
This commit is contained in:
commit
8e261d12e3
|
@ -180,6 +180,89 @@ impl TcpStream {
|
||||||
pub fn write_timeout(&self) -> io::Result<Option<Duration>> {
|
pub fn write_timeout(&self) -> io::Result<Option<Duration>> {
|
||||||
self.0.write_timeout()
|
self.0.write_timeout()
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// Sets the value of the `TCP_NODELAY` option on this socket.
|
||||||
|
///
|
||||||
|
/// If set, this option disables the Nagle algorithm. This means that
|
||||||
|
/// segments are always sent as soon as possible, even if there is only a
|
||||||
|
/// small amount of data. When not set, data is buffered until there is a
|
||||||
|
/// sufficient amount to send out, thereby avoiding the frequent sending of
|
||||||
|
/// small packets.
|
||||||
|
#[stable(feature = "net2_mutators", since = "1.9.0")]
|
||||||
|
pub fn set_nodelay(&self, nodelay: bool) -> io::Result<()> {
|
||||||
|
self.0.set_nodelay(nodelay)
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Gets the value of the `TCP_NODELAY` option on this socket.
|
||||||
|
///
|
||||||
|
/// For more information about this option, see [`set_nodelay`][link].
|
||||||
|
///
|
||||||
|
/// [link]: #tymethod.set_nodelay
|
||||||
|
#[stable(feature = "net2_mutators", since = "1.9.0")]
|
||||||
|
pub fn nodelay(&self) -> io::Result<bool> {
|
||||||
|
self.0.nodelay()
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Sets the value for the `IP_TTL` option on this socket.
|
||||||
|
///
|
||||||
|
/// This value sets the time-to-live field that is used in every packet sent
|
||||||
|
/// from this socket.
|
||||||
|
#[stable(feature = "net2_mutators", since = "1.9.0")]
|
||||||
|
pub fn set_ttl(&self, ttl: u32) -> io::Result<()> {
|
||||||
|
self.0.set_ttl(ttl)
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Gets the value of the `IP_TTL` option for this socket.
|
||||||
|
///
|
||||||
|
/// For more information about this option, see [`set_ttl`][link].
|
||||||
|
///
|
||||||
|
/// [link]: #tymethod.set_ttl
|
||||||
|
#[stable(feature = "net2_mutators", since = "1.9.0")]
|
||||||
|
pub fn ttl(&self) -> io::Result<u32> {
|
||||||
|
self.0.ttl()
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Sets the value for the `IPV6_V6ONLY` option on this socket.
|
||||||
|
///
|
||||||
|
/// If this is set to `true` then the socket is restricted to sending and
|
||||||
|
/// receiving IPv6 packets only. If this is the case, an IPv4 and an IPv6
|
||||||
|
/// application can each bind the same port at the same time.
|
||||||
|
///
|
||||||
|
/// If this is set to `false` then the socket can be used to send and
|
||||||
|
/// receive packets from an IPv4-mapped IPv6 address.
|
||||||
|
#[stable(feature = "net2_mutators", since = "1.9.0")]
|
||||||
|
pub fn set_only_v6(&self, only_v6: bool) -> io::Result<()> {
|
||||||
|
self.0.set_only_v6(only_v6)
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Gets the value of the `IPV6_V6ONLY` option for this socket.
|
||||||
|
///
|
||||||
|
/// For more information about this option, see [`set_only_v6`][link].
|
||||||
|
///
|
||||||
|
/// [link]: #tymethod.set_only_v6
|
||||||
|
#[stable(feature = "net2_mutators", since = "1.9.0")]
|
||||||
|
pub fn only_v6(&self) -> io::Result<bool> {
|
||||||
|
self.0.only_v6()
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Get the value of the `SO_ERROR` option on this socket.
|
||||||
|
///
|
||||||
|
/// This will retrieve the stored error in the underlying socket, clearing
|
||||||
|
/// the field in the process. This can be useful for checking errors between
|
||||||
|
/// calls.
|
||||||
|
#[stable(feature = "net2_mutators", since = "1.9.0")]
|
||||||
|
pub fn take_error(&self) -> io::Result<Option<io::Error>> {
|
||||||
|
self.0.take_error()
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Moves this TCP stream into or out of nonblocking mode.
|
||||||
|
///
|
||||||
|
/// On Unix this corresponds to calling fcntl, and on Windows this
|
||||||
|
/// corresponds to calling ioctlsocket.
|
||||||
|
#[stable(feature = "net2_mutators", since = "1.9.0")]
|
||||||
|
pub fn set_nonblocking(&self, nonblocking: bool) -> io::Result<()> {
|
||||||
|
self.0.set_nonblocking(nonblocking)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
#[stable(feature = "rust1", since = "1.0.0")]
|
#[stable(feature = "rust1", since = "1.0.0")]
|
||||||
|
@ -278,6 +361,67 @@ impl TcpListener {
|
||||||
pub fn incoming(&self) -> Incoming {
|
pub fn incoming(&self) -> Incoming {
|
||||||
Incoming { listener: self }
|
Incoming { listener: self }
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// Sets the value for the `IP_TTL` option on this socket.
|
||||||
|
///
|
||||||
|
/// This value sets the time-to-live field that is used in every packet sent
|
||||||
|
/// from this socket.
|
||||||
|
#[stable(feature = "net2_mutators", since = "1.9.0")]
|
||||||
|
pub fn set_ttl(&self, ttl: u32) -> io::Result<()> {
|
||||||
|
self.0.set_ttl(ttl)
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Gets the value of the `IP_TTL` option for this socket.
|
||||||
|
///
|
||||||
|
/// For more information about this option, see [`set_ttl`][link].
|
||||||
|
///
|
||||||
|
/// [link]: #tymethod.set_ttl
|
||||||
|
#[stable(feature = "net2_mutators", since = "1.9.0")]
|
||||||
|
pub fn ttl(&self) -> io::Result<u32> {
|
||||||
|
self.0.ttl()
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Sets the value for the `IPV6_V6ONLY` option on this socket.
|
||||||
|
///
|
||||||
|
/// If this is set to `true` then the socket is restricted to sending and
|
||||||
|
/// receiving IPv6 packets only. In this case two IPv4 and IPv6 applications
|
||||||
|
/// can bind the same port at the same time.
|
||||||
|
///
|
||||||
|
/// If this is set to `false` then the socket can be used to send and
|
||||||
|
/// receive packets from an IPv4-mapped IPv6 address.
|
||||||
|
#[stable(feature = "net2_mutators", since = "1.9.0")]
|
||||||
|
pub fn set_only_v6(&self, only_v6: bool) -> io::Result<()> {
|
||||||
|
self.0.set_only_v6(only_v6)
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Gets the value of the `IPV6_V6ONLY` option for this socket.
|
||||||
|
///
|
||||||
|
/// For more information about this option, see [`set_only_v6`][link].
|
||||||
|
///
|
||||||
|
/// [link]: #tymethod.set_only_v6
|
||||||
|
#[stable(feature = "net2_mutators", since = "1.9.0")]
|
||||||
|
pub fn only_v6(&self) -> io::Result<bool> {
|
||||||
|
self.0.only_v6()
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Get the value of the `SO_ERROR` option on this socket.
|
||||||
|
///
|
||||||
|
/// This will retrieve the stored error in the underlying socket, clearing
|
||||||
|
/// the field in the process. This can be useful for checking errors between
|
||||||
|
/// calls.
|
||||||
|
#[stable(feature = "net2_mutators", since = "1.9.0")]
|
||||||
|
pub fn take_error(&self) -> io::Result<Option<io::Error>> {
|
||||||
|
self.0.take_error()
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Moves this TCP stream into or out of nonblocking mode.
|
||||||
|
///
|
||||||
|
/// On Unix this corresponds to calling fcntl, and on Windows this
|
||||||
|
/// corresponds to calling ioctlsocket.
|
||||||
|
#[stable(feature = "net2_mutators", since = "1.9.0")]
|
||||||
|
pub fn set_nonblocking(&self, nonblocking: bool) -> io::Result<()> {
|
||||||
|
self.0.set_nonblocking(nonblocking)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
#[stable(feature = "rust1", since = "1.0.0")]
|
#[stable(feature = "rust1", since = "1.0.0")]
|
||||||
|
@ -969,4 +1113,55 @@ mod tests {
|
||||||
assert!(start.elapsed() > Duration::from_millis(400));
|
assert!(start.elapsed() > Duration::from_millis(400));
|
||||||
drop(listener);
|
drop(listener);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn nodelay() {
|
||||||
|
let addr = next_test_ip4();
|
||||||
|
let _listener = t!(TcpListener::bind(&addr));
|
||||||
|
|
||||||
|
let stream = t!(TcpStream::connect(&("localhost", addr.port())));
|
||||||
|
|
||||||
|
assert_eq!(false, t!(stream.nodelay()));
|
||||||
|
t!(stream.set_nodelay(true));
|
||||||
|
assert_eq!(true, t!(stream.nodelay()));
|
||||||
|
t!(stream.set_nodelay(false));
|
||||||
|
assert_eq!(false, t!(stream.nodelay()));
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn ttl() {
|
||||||
|
let ttl = 100;
|
||||||
|
|
||||||
|
let addr = next_test_ip4();
|
||||||
|
let listener = t!(TcpListener::bind(&addr));
|
||||||
|
|
||||||
|
t!(listener.set_ttl(ttl));
|
||||||
|
assert_eq!(ttl, t!(listener.ttl()));
|
||||||
|
|
||||||
|
let stream = t!(TcpStream::connect(&("localhost", addr.port())));
|
||||||
|
|
||||||
|
t!(stream.set_ttl(ttl));
|
||||||
|
assert_eq!(ttl, t!(stream.ttl()));
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn set_nonblocking() {
|
||||||
|
let addr = next_test_ip4();
|
||||||
|
let listener = t!(TcpListener::bind(&addr));
|
||||||
|
|
||||||
|
t!(listener.set_nonblocking(true));
|
||||||
|
t!(listener.set_nonblocking(false));
|
||||||
|
|
||||||
|
let mut stream = t!(TcpStream::connect(&("localhost", addr.port())));
|
||||||
|
|
||||||
|
t!(stream.set_nonblocking(false));
|
||||||
|
t!(stream.set_nonblocking(true));
|
||||||
|
|
||||||
|
let mut buf = [0];
|
||||||
|
match stream.read(&mut buf) {
|
||||||
|
Ok(_) => panic!("expected error"),
|
||||||
|
Err(ref e) if e.kind() == ErrorKind::WouldBlock => {}
|
||||||
|
Err(e) => panic!("unexpected error {}", e),
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
|
@ -10,7 +10,7 @@
|
||||||
|
|
||||||
use fmt;
|
use fmt;
|
||||||
use io::{self, Error, ErrorKind};
|
use io::{self, Error, ErrorKind};
|
||||||
use net::{ToSocketAddrs, SocketAddr};
|
use net::{ToSocketAddrs, SocketAddr, Ipv4Addr, Ipv6Addr};
|
||||||
use sys_common::net as net_imp;
|
use sys_common::net as net_imp;
|
||||||
use sys_common::{AsInner, FromInner, IntoInner};
|
use sys_common::{AsInner, FromInner, IntoInner};
|
||||||
use time::Duration;
|
use time::Duration;
|
||||||
|
@ -140,6 +140,221 @@ impl UdpSocket {
|
||||||
pub fn write_timeout(&self) -> io::Result<Option<Duration>> {
|
pub fn write_timeout(&self) -> io::Result<Option<Duration>> {
|
||||||
self.0.write_timeout()
|
self.0.write_timeout()
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// Sets the value of the `SO_BROADCAST` option for this socket.
|
||||||
|
///
|
||||||
|
/// When enabled, this socket is allowed to send packets to a broadcast
|
||||||
|
/// address.
|
||||||
|
#[stable(feature = "net2_mutators", since = "1.9.0")]
|
||||||
|
pub fn set_broadcast(&self, broadcast: bool) -> io::Result<()> {
|
||||||
|
self.0.set_broadcast(broadcast)
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Gets the value of the `SO_BROADCAST` option for this socket.
|
||||||
|
///
|
||||||
|
/// For more information about this option, see
|
||||||
|
/// [`set_broadcast`][link].
|
||||||
|
///
|
||||||
|
/// [link]: #tymethod.set_broadcast
|
||||||
|
#[stable(feature = "net2_mutators", since = "1.9.0")]
|
||||||
|
pub fn broadcast(&self) -> io::Result<bool> {
|
||||||
|
self.0.broadcast()
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Sets the value of the `IP_MULTICAST_LOOP` option for this socket.
|
||||||
|
///
|
||||||
|
/// If enabled, multicast packets will be looped back to the local socket.
|
||||||
|
/// Note that this may not have any affect on IPv6 sockets.
|
||||||
|
#[stable(feature = "net2_mutators", since = "1.9.0")]
|
||||||
|
pub fn set_multicast_loop_v4(&self, multicast_loop_v4: bool) -> io::Result<()> {
|
||||||
|
self.0.set_multicast_loop_v4(multicast_loop_v4)
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Gets the value of the `IP_MULTICAST_LOOP` option for this socket.
|
||||||
|
///
|
||||||
|
/// For more information about this option, see
|
||||||
|
/// [`set_multicast_loop_v4`][link].
|
||||||
|
///
|
||||||
|
/// [link]: #tymethod.set_multicast_loop_v4
|
||||||
|
#[stable(feature = "net2_mutators", since = "1.9.0")]
|
||||||
|
pub fn multicast_loop_v4(&self) -> io::Result<bool> {
|
||||||
|
self.0.multicast_loop_v4()
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Sets the value of the `IP_MULTICAST_TTL` option for this socket.
|
||||||
|
///
|
||||||
|
/// Indicates the time-to-live value of outgoing multicast packets for
|
||||||
|
/// this socket. The default value is 1 which means that multicast packets
|
||||||
|
/// don't leave the local network unless explicitly requested.
|
||||||
|
///
|
||||||
|
/// Note that this may not have any affect on IPv6 sockets.
|
||||||
|
#[stable(feature = "net2_mutators", since = "1.9.0")]
|
||||||
|
pub fn set_multicast_ttl_v4(&self, multicast_ttl_v4: u32) -> io::Result<()> {
|
||||||
|
self.0.set_multicast_ttl_v4(multicast_ttl_v4)
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Gets the value of the `IP_MULTICAST_TTL` option for this socket.
|
||||||
|
///
|
||||||
|
/// For more information about this option, see
|
||||||
|
/// [`set_multicast_ttl_v4`][link].
|
||||||
|
///
|
||||||
|
/// [link]: #tymethod.set_multicast_ttl_v4
|
||||||
|
#[stable(feature = "net2_mutators", since = "1.9.0")]
|
||||||
|
pub fn multicast_ttl_v4(&self) -> io::Result<u32> {
|
||||||
|
self.0.multicast_ttl_v4()
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Sets the value of the `IPV6_MULTICAST_LOOP` option for this socket.
|
||||||
|
///
|
||||||
|
/// Controls whether this socket sees the multicast packets it sends itself.
|
||||||
|
/// Note that this may not have any affect on IPv4 sockets.
|
||||||
|
#[stable(feature = "net2_mutators", since = "1.9.0")]
|
||||||
|
pub fn set_multicast_loop_v6(&self, multicast_loop_v6: bool) -> io::Result<()> {
|
||||||
|
self.0.set_multicast_loop_v6(multicast_loop_v6)
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Gets the value of the `IPV6_MULTICAST_LOOP` option for this socket.
|
||||||
|
///
|
||||||
|
/// For more information about this option, see
|
||||||
|
/// [`set_multicast_loop_v6`][link].
|
||||||
|
///
|
||||||
|
/// [link]: #tymethod.set_multicast_loop_v6
|
||||||
|
#[stable(feature = "net2_mutators", since = "1.9.0")]
|
||||||
|
pub fn multicast_loop_v6(&self) -> io::Result<bool> {
|
||||||
|
self.0.multicast_loop_v6()
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Sets the value for the `IP_TTL` option on this socket.
|
||||||
|
///
|
||||||
|
/// This value sets the time-to-live field that is used in every packet sent
|
||||||
|
/// from this socket.
|
||||||
|
#[stable(feature = "net2_mutators", since = "1.9.0")]
|
||||||
|
pub fn set_ttl(&self, ttl: u32) -> io::Result<()> {
|
||||||
|
self.0.set_ttl(ttl)
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Gets the value of the `IP_TTL` option for this socket.
|
||||||
|
///
|
||||||
|
/// For more information about this option, see [`set_ttl`][link].
|
||||||
|
///
|
||||||
|
/// [link]: #tymethod.set_ttl
|
||||||
|
#[stable(feature = "net2_mutators", since = "1.9.0")]
|
||||||
|
pub fn ttl(&self) -> io::Result<u32> {
|
||||||
|
self.0.ttl()
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Sets the value for the `IPV6_V6ONLY` option on this socket.
|
||||||
|
///
|
||||||
|
/// If this is set to `true` then the socket is restricted to sending and
|
||||||
|
/// receiving IPv6 packets only. If this is the case, an IPv4 and an IPv6
|
||||||
|
/// application can each bind the same port at the same time.
|
||||||
|
///
|
||||||
|
/// If this is set to `false` then the socket can be used to send and
|
||||||
|
/// receive packets from an IPv4-mapped IPv6 address.
|
||||||
|
#[stable(feature = "net2_mutators", since = "1.9.0")]
|
||||||
|
pub fn set_only_v6(&self, only_v6: bool) -> io::Result<()> {
|
||||||
|
self.0.set_only_v6(only_v6)
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Gets the value of the `IPV6_V6ONLY` option for this socket.
|
||||||
|
///
|
||||||
|
/// For more information about this option, see [`set_only_v6`][link].
|
||||||
|
///
|
||||||
|
/// [link]: #tymethod.set_only_v6
|
||||||
|
#[stable(feature = "net2_mutators", since = "1.9.0")]
|
||||||
|
pub fn only_v6(&self) -> io::Result<bool> {
|
||||||
|
self.0.only_v6()
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Executes an operation of the `IP_ADD_MEMBERSHIP` type.
|
||||||
|
///
|
||||||
|
/// This function specifies a new multicast group for this socket to join.
|
||||||
|
/// The address must be a valid multicast address, and `interface` is the
|
||||||
|
/// address of the local interface with which the system should join the
|
||||||
|
/// multicast group. If it's equal to `INADDR_ANY` then an appropriate
|
||||||
|
/// interface is chosen by the system.
|
||||||
|
#[stable(feature = "net2_mutators", since = "1.9.0")]
|
||||||
|
pub fn join_multicast_v4(&self, multiaddr: &Ipv4Addr, interface: &Ipv4Addr) -> io::Result<()> {
|
||||||
|
self.0.join_multicast_v4(multiaddr, interface)
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Executes an operation of the `IPV6_ADD_MEMBERSHIP` type.
|
||||||
|
///
|
||||||
|
/// This function specifies a new multicast group for this socket to join.
|
||||||
|
/// The address must be a valid multicast address, and `interface` is the
|
||||||
|
/// index of the interface to join/leave (or 0 to indicate any interface).
|
||||||
|
#[stable(feature = "net2_mutators", since = "1.9.0")]
|
||||||
|
pub fn join_multicast_v6(&self, multiaddr: &Ipv6Addr, interface: u32) -> io::Result<()> {
|
||||||
|
self.0.join_multicast_v6(multiaddr, interface)
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Executes an operation of the `IP_DROP_MEMBERSHIP` type.
|
||||||
|
///
|
||||||
|
/// For more information about this option, see
|
||||||
|
/// [`join_multicast_v4`][link].
|
||||||
|
///
|
||||||
|
/// [link]: #tymethod.join_multicast_v4
|
||||||
|
#[stable(feature = "net2_mutators", since = "1.9.0")]
|
||||||
|
pub fn leave_multicast_v4(&self, multiaddr: &Ipv4Addr, interface: &Ipv4Addr) -> io::Result<()> {
|
||||||
|
self.0.leave_multicast_v4(multiaddr, interface)
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Executes an operation of the `IPV6_DROP_MEMBERSHIP` type.
|
||||||
|
///
|
||||||
|
/// For more information about this option, see
|
||||||
|
/// [`join_multicast_v6`][link].
|
||||||
|
///
|
||||||
|
/// [link]: #tymethod.join_multicast_v6
|
||||||
|
#[stable(feature = "net2_mutators", since = "1.9.0")]
|
||||||
|
pub fn leave_multicast_v6(&self, multiaddr: &Ipv6Addr, interface: u32) -> io::Result<()> {
|
||||||
|
self.0.leave_multicast_v6(multiaddr, interface)
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Get the value of the `SO_ERROR` option on this socket.
|
||||||
|
///
|
||||||
|
/// This will retrieve the stored error in the underlying socket, clearing
|
||||||
|
/// the field in the process. This can be useful for checking errors between
|
||||||
|
/// calls.
|
||||||
|
#[stable(feature = "net2_mutators", since = "1.9.0")]
|
||||||
|
pub fn take_error(&self) -> io::Result<Option<io::Error>> {
|
||||||
|
self.0.take_error()
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Connects this UDP socket to a remote address, allowing the `send` and
|
||||||
|
/// `recv` syscalls to be used to send data and also applies filters to only
|
||||||
|
/// receive data from the specified address.
|
||||||
|
#[stable(feature = "net2_mutators", since = "1.9.0")]
|
||||||
|
pub fn connect<A: ToSocketAddrs>(&self, addr: A) -> io::Result<()> {
|
||||||
|
super::each_addr(addr, |addr| self.0.connect(addr))
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Sends data on the socket to the remote address to which it is connected.
|
||||||
|
///
|
||||||
|
/// The `connect` method will connect this socket to a remote address. This
|
||||||
|
/// method will fail if the socket is not connected.
|
||||||
|
#[stable(feature = "net2_mutators", since = "1.9.0")]
|
||||||
|
pub fn send(&self, buf: &[u8]) -> io::Result<usize> {
|
||||||
|
self.0.send(buf)
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Receives data on the socket from the remote address to which it is
|
||||||
|
/// connected.
|
||||||
|
///
|
||||||
|
/// The `connect` method will connect this socket to a remote address. This
|
||||||
|
/// method will fail if the socket is not connected.
|
||||||
|
#[stable(feature = "net2_mutators", since = "1.9.0")]
|
||||||
|
pub fn recv(&self, buf: &mut [u8]) -> io::Result<usize> {
|
||||||
|
self.0.recv(buf)
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Moves this TCP stream into or out of nonblocking mode.
|
||||||
|
///
|
||||||
|
/// On Unix this corresponds to calling fcntl, and on Windows this
|
||||||
|
/// corresponds to calling ioctlsocket.
|
||||||
|
#[stable(feature = "net2_mutators", since = "1.9.0")]
|
||||||
|
pub fn set_nonblocking(&self, nonblocking: bool) -> io::Result<()> {
|
||||||
|
self.0.set_nonblocking(nonblocking)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
impl AsInner<net_imp::UdpSocket> for UdpSocket {
|
impl AsInner<net_imp::UdpSocket> for UdpSocket {
|
||||||
|
@ -400,4 +615,40 @@ mod tests {
|
||||||
assert!(kind == ErrorKind::WouldBlock || kind == ErrorKind::TimedOut);
|
assert!(kind == ErrorKind::WouldBlock || kind == ErrorKind::TimedOut);
|
||||||
assert!(start.elapsed() > Duration::from_millis(400));
|
assert!(start.elapsed() > Duration::from_millis(400));
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn connect_send_recv() {
|
||||||
|
let addr = next_test_ip4();
|
||||||
|
|
||||||
|
let socket = t!(UdpSocket::bind(&addr));
|
||||||
|
t!(socket.connect(addr));
|
||||||
|
|
||||||
|
t!(socket.send(b"hello world"));
|
||||||
|
|
||||||
|
let mut buf = [0; 11];
|
||||||
|
t!(socket.recv(&mut buf));
|
||||||
|
assert_eq!(b"hello world", &buf[..]);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn ttl() {
|
||||||
|
let ttl = 100;
|
||||||
|
|
||||||
|
let addr = next_test_ip4();
|
||||||
|
|
||||||
|
let stream = t!(UdpSocket::bind(&addr));
|
||||||
|
|
||||||
|
t!(stream.set_ttl(ttl));
|
||||||
|
assert_eq!(ttl, t!(stream.ttl()));
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn set_nonblocking() {
|
||||||
|
let addr = next_test_ip4();
|
||||||
|
|
||||||
|
let stream = t!(UdpSocket::bind(&addr));
|
||||||
|
|
||||||
|
t!(stream.set_nonblocking(true));
|
||||||
|
t!(stream.set_nonblocking(false));
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
|
@ -17,7 +17,7 @@ use io::{self, Error, ErrorKind};
|
||||||
use libc::{c_int, c_char, c_void};
|
use libc::{c_int, c_char, c_void};
|
||||||
use mem;
|
use mem;
|
||||||
#[allow(deprecated)]
|
#[allow(deprecated)]
|
||||||
use net::{SocketAddr, Shutdown, IpAddr};
|
use net::{SocketAddr, Shutdown, IpAddr, Ipv4Addr, Ipv6Addr};
|
||||||
use ptr;
|
use ptr;
|
||||||
use str::from_utf8;
|
use str::from_utf8;
|
||||||
use sys::net::{cvt, cvt_r, cvt_gai, Socket, init, wrlen_t};
|
use sys::net::{cvt, cvt_r, cvt_gai, Socket, init, wrlen_t};
|
||||||
|
@ -25,6 +25,23 @@ use sys::net::netc as c;
|
||||||
use sys_common::{AsInner, FromInner, IntoInner};
|
use sys_common::{AsInner, FromInner, IntoInner};
|
||||||
use time::Duration;
|
use time::Duration;
|
||||||
|
|
||||||
|
#[cfg(any(target_os = "dragonfly", target_os = "freebsd",
|
||||||
|
target_os = "ios", target_os = "macos",
|
||||||
|
target_os = "openbsd", target_os = "netbsd"))]
|
||||||
|
use sys::net::netc::IPV6_JOIN_GROUP as IPV6_ADD_MEMBERSHIP;
|
||||||
|
#[cfg(not(any(target_os = "dragonfly", target_os = "freebsd",
|
||||||
|
target_os = "ios", target_os = "macos",
|
||||||
|
target_os = "openbsd", target_os = "netbsd")))]
|
||||||
|
use sys::net::netc::IPV6_ADD_MEMBERSHIP;
|
||||||
|
#[cfg(any(target_os = "dragonfly", target_os = "freebsd",
|
||||||
|
target_os = "ios", target_os = "macos",
|
||||||
|
target_os = "openbsd", target_os = "netbsd"))]
|
||||||
|
use sys::net::netc::IPV6_LEAVE_GROUP as IPV6_DROP_MEMBERSHIP;
|
||||||
|
#[cfg(not(any(target_os = "dragonfly", target_os = "freebsd",
|
||||||
|
target_os = "ios", target_os = "macos",
|
||||||
|
target_os = "openbsd", target_os = "netbsd")))]
|
||||||
|
use sys::net::netc::IPV6_DROP_MEMBERSHIP;
|
||||||
|
|
||||||
////////////////////////////////////////////////////////////////////////////////
|
////////////////////////////////////////////////////////////////////////////////
|
||||||
// sockaddr and misc bindings
|
// sockaddr and misc bindings
|
||||||
////////////////////////////////////////////////////////////////////////////////
|
////////////////////////////////////////////////////////////////////////////////
|
||||||
|
@ -84,6 +101,16 @@ fn sockaddr_to_addr(storage: &c::sockaddr_storage,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[cfg(target_os = "android")]
|
||||||
|
fn to_ipv6mr_interface(value: u32) -> c_int {
|
||||||
|
value as c_int
|
||||||
|
}
|
||||||
|
|
||||||
|
#[cfg(not(target_os = "android"))]
|
||||||
|
fn to_ipv6mr_interface(value: u32) -> ::libc::c_uint {
|
||||||
|
value as ::libc::c_uint
|
||||||
|
}
|
||||||
|
|
||||||
////////////////////////////////////////////////////////////////////////////////
|
////////////////////////////////////////////////////////////////////////////////
|
||||||
// get_host_addresses
|
// get_host_addresses
|
||||||
////////////////////////////////////////////////////////////////////////////////
|
////////////////////////////////////////////////////////////////////////////////
|
||||||
|
@ -228,6 +255,45 @@ impl TcpStream {
|
||||||
pub fn duplicate(&self) -> io::Result<TcpStream> {
|
pub fn duplicate(&self) -> io::Result<TcpStream> {
|
||||||
self.inner.duplicate().map(|s| TcpStream { inner: s })
|
self.inner.duplicate().map(|s| TcpStream { inner: s })
|
||||||
}
|
}
|
||||||
|
|
||||||
|
pub fn set_nodelay(&self, nodelay: bool) -> io::Result<()> {
|
||||||
|
self.inner.set_nodelay(nodelay)
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn nodelay(&self) -> io::Result<bool> {
|
||||||
|
self.inner.nodelay()
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn set_ttl(&self, ttl: u32) -> io::Result<()> {
|
||||||
|
setsockopt(&self.inner, c::IPPROTO_IP, c::IP_TTL, ttl as c_int)
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn ttl(&self) -> io::Result<u32> {
|
||||||
|
let raw: c_int = try!(getsockopt(&self.inner, c::IPPROTO_IP, c::IP_TTL));
|
||||||
|
Ok(raw as u32)
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn set_only_v6(&self, only_v6: bool) -> io::Result<()> {
|
||||||
|
setsockopt(&self.inner, c::IPPROTO_IPV6, c::IPV6_V6ONLY, only_v6 as c_int)
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn only_v6(&self) -> io::Result<bool> {
|
||||||
|
let raw: c_int = try!(getsockopt(&self.inner, c::IPPROTO_IPV6, c::IPV6_V6ONLY));
|
||||||
|
Ok(raw != 0)
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn take_error(&self) -> io::Result<Option<io::Error>> {
|
||||||
|
let raw: c_int = try!(getsockopt(&self.inner, c::SOL_SOCKET, c::SO_ERROR));
|
||||||
|
if raw == 0 {
|
||||||
|
Ok(None)
|
||||||
|
} else {
|
||||||
|
Ok(Some(io::Error::from_raw_os_error(raw as i32)))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn set_nonblocking(&self, nonblocking: bool) -> io::Result<()> {
|
||||||
|
self.inner.set_nonblocking(nonblocking)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
impl FromInner<Socket> for TcpStream {
|
impl FromInner<Socket> for TcpStream {
|
||||||
|
@ -307,6 +373,37 @@ impl TcpListener {
|
||||||
pub fn duplicate(&self) -> io::Result<TcpListener> {
|
pub fn duplicate(&self) -> io::Result<TcpListener> {
|
||||||
self.inner.duplicate().map(|s| TcpListener { inner: s })
|
self.inner.duplicate().map(|s| TcpListener { inner: s })
|
||||||
}
|
}
|
||||||
|
|
||||||
|
pub fn set_ttl(&self, ttl: u32) -> io::Result<()> {
|
||||||
|
setsockopt(&self.inner, c::IPPROTO_IP, c::IP_TTL, ttl as c_int)
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn ttl(&self) -> io::Result<u32> {
|
||||||
|
let raw: c_int = try!(getsockopt(&self.inner, c::IPPROTO_IP, c::IP_TTL));
|
||||||
|
Ok(raw as u32)
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn set_only_v6(&self, only_v6: bool) -> io::Result<()> {
|
||||||
|
setsockopt(&self.inner, c::IPPROTO_IPV6, c::IPV6_V6ONLY, only_v6 as c_int)
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn only_v6(&self) -> io::Result<bool> {
|
||||||
|
let raw: c_int = try!(getsockopt(&self.inner, c::IPPROTO_IPV6, c::IPV6_V6ONLY));
|
||||||
|
Ok(raw != 0)
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn take_error(&self) -> io::Result<Option<io::Error>> {
|
||||||
|
let raw: c_int = try!(getsockopt(&self.inner, c::SOL_SOCKET, c::SO_ERROR));
|
||||||
|
if raw == 0 {
|
||||||
|
Ok(None)
|
||||||
|
} else {
|
||||||
|
Ok(Some(io::Error::from_raw_os_error(raw as i32)))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn set_nonblocking(&self, nonblocking: bool) -> io::Result<()> {
|
||||||
|
self.inner.set_nonblocking(nonblocking)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
impl FromInner<Socket> for TcpListener {
|
impl FromInner<Socket> for TcpListener {
|
||||||
|
@ -401,6 +498,129 @@ impl UdpSocket {
|
||||||
pub fn write_timeout(&self) -> io::Result<Option<Duration>> {
|
pub fn write_timeout(&self) -> io::Result<Option<Duration>> {
|
||||||
self.inner.timeout(c::SO_SNDTIMEO)
|
self.inner.timeout(c::SO_SNDTIMEO)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
pub fn set_broadcast(&self, broadcast: bool) -> io::Result<()> {
|
||||||
|
setsockopt(&self.inner, c::SOL_SOCKET, c::SO_BROADCAST, broadcast as c_int)
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn broadcast(&self) -> io::Result<bool> {
|
||||||
|
let raw: c_int = try!(getsockopt(&self.inner, c::SOL_SOCKET, c::SO_BROADCAST));
|
||||||
|
Ok(raw != 0)
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn set_multicast_loop_v4(&self, multicast_loop_v4: bool) -> io::Result<()> {
|
||||||
|
setsockopt(&self.inner, c::IPPROTO_IP, c::IP_MULTICAST_LOOP, multicast_loop_v4 as c_int)
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn multicast_loop_v4(&self) -> io::Result<bool> {
|
||||||
|
let raw: c_int = try!(getsockopt(&self.inner, c::IPPROTO_IP, c::IP_MULTICAST_LOOP));
|
||||||
|
Ok(raw != 0)
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn set_multicast_ttl_v4(&self, multicast_ttl_v4: u32) -> io::Result<()> {
|
||||||
|
setsockopt(&self.inner, c::IPPROTO_IP, c::IP_MULTICAST_TTL, multicast_ttl_v4 as c_int)
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn multicast_ttl_v4(&self) -> io::Result<u32> {
|
||||||
|
let raw: c_int = try!(getsockopt(&self.inner, c::IPPROTO_IP, c::IP_MULTICAST_TTL));
|
||||||
|
Ok(raw as u32)
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn set_multicast_loop_v6(&self, multicast_loop_v6: bool) -> io::Result<()> {
|
||||||
|
setsockopt(&self.inner, c::IPPROTO_IPV6, c::IPV6_MULTICAST_LOOP, multicast_loop_v6 as c_int)
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn multicast_loop_v6(&self) -> io::Result<bool> {
|
||||||
|
let raw: c_int = try!(getsockopt(&self.inner, c::IPPROTO_IPV6, c::IPV6_MULTICAST_LOOP));
|
||||||
|
Ok(raw != 0)
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn join_multicast_v4(&self, multiaddr: &Ipv4Addr, interface: &Ipv4Addr)
|
||||||
|
-> io::Result<()> {
|
||||||
|
let mreq = c::ip_mreq {
|
||||||
|
imr_multiaddr: *multiaddr.as_inner(),
|
||||||
|
imr_interface: *interface.as_inner(),
|
||||||
|
};
|
||||||
|
setsockopt(&self.inner, c::IPPROTO_IP, c::IP_ADD_MEMBERSHIP, mreq)
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn join_multicast_v6(&self, multiaddr: &Ipv6Addr, interface: u32)
|
||||||
|
-> io::Result<()> {
|
||||||
|
let mreq = c::ipv6_mreq {
|
||||||
|
ipv6mr_multiaddr: *multiaddr.as_inner(),
|
||||||
|
ipv6mr_interface: to_ipv6mr_interface(interface),
|
||||||
|
};
|
||||||
|
setsockopt(&self.inner, c::IPPROTO_IPV6, IPV6_ADD_MEMBERSHIP, mreq)
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn leave_multicast_v4(&self, multiaddr: &Ipv4Addr, interface: &Ipv4Addr)
|
||||||
|
-> io::Result<()> {
|
||||||
|
let mreq = c::ip_mreq {
|
||||||
|
imr_multiaddr: *multiaddr.as_inner(),
|
||||||
|
imr_interface: *interface.as_inner(),
|
||||||
|
};
|
||||||
|
setsockopt(&self.inner, c::IPPROTO_IP, c::IP_DROP_MEMBERSHIP, mreq)
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn leave_multicast_v6(&self, multiaddr: &Ipv6Addr, interface: u32)
|
||||||
|
-> io::Result<()> {
|
||||||
|
let mreq = c::ipv6_mreq {
|
||||||
|
ipv6mr_multiaddr: *multiaddr.as_inner(),
|
||||||
|
ipv6mr_interface: to_ipv6mr_interface(interface),
|
||||||
|
};
|
||||||
|
setsockopt(&self.inner, c::IPPROTO_IPV6, IPV6_DROP_MEMBERSHIP, mreq)
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn set_ttl(&self, ttl: u32) -> io::Result<()> {
|
||||||
|
setsockopt(&self.inner, c::IPPROTO_IP, c::IP_TTL, ttl as c_int)
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn ttl(&self) -> io::Result<u32> {
|
||||||
|
let raw: c_int = try!(getsockopt(&self.inner, c::IPPROTO_IP, c::IP_TTL));
|
||||||
|
Ok(raw as u32)
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn set_only_v6(&self, only_v6: bool) -> io::Result<()> {
|
||||||
|
setsockopt(&self.inner, c::IPPROTO_IPV6, c::IPV6_V6ONLY, only_v6 as c_int)
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn only_v6(&self) -> io::Result<bool> {
|
||||||
|
let raw: c_int = try!(getsockopt(&self.inner, c::IPPROTO_IPV6, c::IPV6_V6ONLY));
|
||||||
|
Ok(raw != 0)
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn take_error(&self) -> io::Result<Option<io::Error>> {
|
||||||
|
let raw: c_int = try!(getsockopt(&self.inner, c::SOL_SOCKET, c::SO_ERROR));
|
||||||
|
if raw == 0 {
|
||||||
|
Ok(None)
|
||||||
|
} else {
|
||||||
|
Ok(Some(io::Error::from_raw_os_error(raw as i32)))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn set_nonblocking(&self, nonblocking: bool) -> io::Result<()> {
|
||||||
|
self.inner.set_nonblocking(nonblocking)
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn recv(&self, buf: &mut [u8]) -> io::Result<usize> {
|
||||||
|
self.inner.read(buf)
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn send(&self, buf: &[u8]) -> io::Result<usize> {
|
||||||
|
let len = cmp::min(buf.len(), <wrlen_t>::max_value() as usize) as wrlen_t;
|
||||||
|
let ret = try!(cvt(unsafe {
|
||||||
|
c::send(*self.inner.as_inner(),
|
||||||
|
buf.as_ptr() as *const c_void,
|
||||||
|
len,
|
||||||
|
0)
|
||||||
|
}));
|
||||||
|
Ok(ret as usize)
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn connect(&self, addr: &SocketAddr) -> io::Result<()> {
|
||||||
|
let (addrp, len) = addr.into_inner();
|
||||||
|
cvt_r(|| unsafe { c::connect(*self.inner.as_inner(), addrp, len) }).map(|_| ())
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
impl FromInner<Socket> for UdpSocket {
|
impl FromInner<Socket> for UdpSocket {
|
||||||
|
|
|
@ -168,6 +168,20 @@ impl Socket {
|
||||||
try!(cvt(unsafe { libc::shutdown(self.0.raw(), how) }));
|
try!(cvt(unsafe { libc::shutdown(self.0.raw(), how) }));
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
||||||
|
pub fn set_nodelay(&self, nodelay: bool) -> io::Result<()> {
|
||||||
|
setsockopt(self, libc::IPPROTO_TCP, libc::TCP_NODELAY, nodelay as c_int)
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn nodelay(&self) -> io::Result<bool> {
|
||||||
|
let raw: c_int = try!(getsockopt(self, libc::IPPROTO_TCP, libc::TCP_NODELAY));
|
||||||
|
Ok(raw != 0)
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn set_nonblocking(&self, nonblocking: bool) -> io::Result<()> {
|
||||||
|
let mut nonblocking = nonblocking as libc::c_ulong;
|
||||||
|
cvt(unsafe { libc::ioctl(*self.as_inner(), libc::FIONBIO, &mut nonblocking) }).map(|_| ())
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
impl AsInner<c_int> for Socket {
|
impl AsInner<c_int> for Socket {
|
||||||
|
|
|
@ -13,7 +13,7 @@
|
||||||
#![allow(bad_style)]
|
#![allow(bad_style)]
|
||||||
#![cfg_attr(test, allow(dead_code))]
|
#![cfg_attr(test, allow(dead_code))]
|
||||||
|
|
||||||
use os::raw::{c_int, c_uint, c_ulong, c_long, c_longlong, c_ushort};
|
use os::raw::{c_int, c_uint, c_ulong, c_long, c_longlong, c_ushort,};
|
||||||
use os::raw::{c_char, c_ulonglong};
|
use os::raw::{c_char, c_ulonglong};
|
||||||
use libc::{wchar_t, size_t, c_void};
|
use libc::{wchar_t, size_t, c_void};
|
||||||
use ptr;
|
use ptr;
|
||||||
|
@ -114,6 +114,8 @@ pub const FILE_FLAG_OPEN_REPARSE_POINT: DWORD = 0x00200000;
|
||||||
pub const FILE_FLAG_BACKUP_SEMANTICS: DWORD = 0x02000000;
|
pub const FILE_FLAG_BACKUP_SEMANTICS: DWORD = 0x02000000;
|
||||||
pub const SECURITY_SQOS_PRESENT: DWORD = 0x00100000;
|
pub const SECURITY_SQOS_PRESENT: DWORD = 0x00100000;
|
||||||
|
|
||||||
|
pub const FIONBIO: c_ulong = 0x8004667e;
|
||||||
|
|
||||||
#[repr(C)]
|
#[repr(C)]
|
||||||
#[derive(Copy)]
|
#[derive(Copy)]
|
||||||
pub struct WIN32_FIND_DATAW {
|
pub struct WIN32_FIND_DATAW {
|
||||||
|
@ -223,6 +225,33 @@ pub const SOL_SOCKET: c_int = 0xffff;
|
||||||
pub const SO_RCVTIMEO: c_int = 0x1006;
|
pub const SO_RCVTIMEO: c_int = 0x1006;
|
||||||
pub const SO_SNDTIMEO: c_int = 0x1005;
|
pub const SO_SNDTIMEO: c_int = 0x1005;
|
||||||
pub const SO_REUSEADDR: c_int = 0x0004;
|
pub const SO_REUSEADDR: c_int = 0x0004;
|
||||||
|
pub const IPPROTO_IP: c_int = 0;
|
||||||
|
pub const IPPROTO_TCP: c_int = 6;
|
||||||
|
pub const IPPROTO_IPV6: c_int = 41;
|
||||||
|
pub const TCP_NODELAY: c_int = 0x0001;
|
||||||
|
pub const IP_TTL: c_int = 4;
|
||||||
|
pub const IPV6_V6ONLY: c_int = 27;
|
||||||
|
pub const SO_ERROR: c_int = 0x1007;
|
||||||
|
pub const SO_BROADCAST: c_int = 0x0020;
|
||||||
|
pub const IP_MULTICAST_LOOP: c_int = 11;
|
||||||
|
pub const IPV6_MULTICAST_LOOP: c_int = 11;
|
||||||
|
pub const IP_MULTICAST_TTL: c_int = 10;
|
||||||
|
pub const IP_ADD_MEMBERSHIP: c_int = 12;
|
||||||
|
pub const IP_DROP_MEMBERSHIP: c_int = 13;
|
||||||
|
pub const IPV6_ADD_MEMBERSHIP: c_int = 12;
|
||||||
|
pub const IPV6_DROP_MEMBERSHIP: c_int = 13;
|
||||||
|
|
||||||
|
#[repr(C)]
|
||||||
|
pub struct ip_mreq {
|
||||||
|
pub imr_multiaddr: in_addr,
|
||||||
|
pub imr_interface: in_addr,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[repr(C)]
|
||||||
|
pub struct ipv6_mreq {
|
||||||
|
pub ipv6mr_multiaddr: in6_addr,
|
||||||
|
pub ipv6mr_interface: c_uint,
|
||||||
|
}
|
||||||
|
|
||||||
pub const VOLUME_NAME_DOS: DWORD = 0x0;
|
pub const VOLUME_NAME_DOS: DWORD = 0x0;
|
||||||
pub const MOVEFILE_REPLACE_EXISTING: DWORD = 1;
|
pub const MOVEFILE_REPLACE_EXISTING: DWORD = 1;
|
||||||
|
@ -833,6 +862,7 @@ extern "system" {
|
||||||
lpProtocolInfo: LPWSAPROTOCOL_INFO,
|
lpProtocolInfo: LPWSAPROTOCOL_INFO,
|
||||||
g: GROUP,
|
g: GROUP,
|
||||||
dwFlags: DWORD) -> SOCKET;
|
dwFlags: DWORD) -> SOCKET;
|
||||||
|
pub fn ioctlsocket(s: SOCKET, cmd: c_long, argp: *mut c_ulong) -> c_int;
|
||||||
pub fn InitializeCriticalSection(CriticalSection: *mut CRITICAL_SECTION);
|
pub fn InitializeCriticalSection(CriticalSection: *mut CRITICAL_SECTION);
|
||||||
pub fn EnterCriticalSection(CriticalSection: *mut CRITICAL_SECTION);
|
pub fn EnterCriticalSection(CriticalSection: *mut CRITICAL_SECTION);
|
||||||
pub fn TryEnterCriticalSection(CriticalSection: *mut CRITICAL_SECTION) -> BOOLEAN;
|
pub fn TryEnterCriticalSection(CriticalSection: *mut CRITICAL_SECTION) -> BOOLEAN;
|
||||||
|
|
|
@ -10,7 +10,7 @@
|
||||||
|
|
||||||
use cmp;
|
use cmp;
|
||||||
use io;
|
use io;
|
||||||
use libc::{c_int, c_void};
|
use libc::{c_int, c_void, c_ulong};
|
||||||
use mem;
|
use mem;
|
||||||
use net::{SocketAddr, Shutdown};
|
use net::{SocketAddr, Shutdown};
|
||||||
use num::One;
|
use num::One;
|
||||||
|
@ -185,6 +185,25 @@ impl Socket {
|
||||||
try!(cvt(unsafe { c::shutdown(self.0, how) }));
|
try!(cvt(unsafe { c::shutdown(self.0, how) }));
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
||||||
|
pub fn set_nonblocking(&self, nonblocking: bool) -> io::Result<()> {
|
||||||
|
let mut nonblocking = nonblocking as c_ulong;
|
||||||
|
let r = unsafe { c::ioctlsocket(self.0, c::FIONBIO as c_int, &mut nonblocking) };
|
||||||
|
if r == 0 {
|
||||||
|
Ok(())
|
||||||
|
} else {
|
||||||
|
Err(io::Error::last_os_error())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn set_nodelay(&self, nodelay: bool) -> io::Result<()> {
|
||||||
|
net::setsockopt(self, c::IPPROTO_TCP, c::TCP_NODELAY, nodelay as c::BYTE)
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn nodelay(&self) -> io::Result<bool> {
|
||||||
|
let raw: c::BYTE = try!(net::getsockopt(self, c::IPPROTO_TCP, c::TCP_NODELAY));
|
||||||
|
Ok(raw != 0)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Drop for Socket {
|
impl Drop for Socket {
|
||||||
|
|
Loading…
Reference in New Issue