| // Copyright (C) 2017-2018 Baidu, Inc. All Rights Reserved. |
| // |
| // Redistribution and use in source and binary forms, with or without |
| // modification, are permitted provided that the following conditions |
| // are met: |
| // |
| // * Redistributions of source code must retain the above copyright |
| // notice, this list of conditions and the following disclaimer. |
| // * Redistributions in binary form must reproduce the above copyright |
| // notice, this list of conditions and the following disclaimer in |
| // the documentation and/or other materials provided with the |
| // distribution. |
| // * Neither the name of Baidu, Inc., nor the names of its |
| // contributors may be used to endorse or promote products derived |
| // from this software without specific prior written permission. |
| // |
| // THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS |
| // "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT |
| // LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR |
| // A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT |
| // OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, |
| // SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT |
| // LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, |
| // DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY |
| // THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
| // (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE |
| // OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
| |
| //! Networking primitives for TCP/UDP communication. |
| //! |
| //! This module provides networking functionality for the Transmission Control and User |
| //! Datagram Protocols, as well as types for IP and socket addresses. |
| //! |
| |
| //use core::fmt; |
| use io::{self, Error, ErrorKind}; |
| //use sys_common::net as net_imp; |
| |
| pub use self::ip::{IpAddr, Ipv4Addr, Ipv6Addr, Ipv6MulticastScope}; |
| pub use self::addr::{SocketAddr, SocketAddrV4, SocketAddrV6, ToSocketAddrs}; |
| #[cfg(feature = "net")] |
| pub use self::tcp::TcpStream; |
| #[cfg(feature = "net")] |
| pub use self::udp::UdpSocket; |
| pub use self::parser::AddrParseError; |
| |
| mod ip; |
| mod addr; |
| mod parser; |
| #[cfg(feature = "net")] |
| mod tcp; |
| #[cfg(feature = "net")] |
| mod udp; |
| |
| /// Possible values which can be passed to the [`shutdown`] method of |
| #[derive(Copy, Clone, PartialEq, Eq, Debug)] |
| pub enum Shutdown { |
| /// The reading portion of the [`TcpStream`] should be shut down. |
| /// |
| /// All currently blocked and future [reads] will return [`Ok(0)`]. |
| /// |
| Read, |
| /// The writing portion of the [`TcpStream`] should be shut down. |
| /// |
| /// All currently blocked and future [writes] will return an error. |
| /// |
| Write, |
| /// Both the reading and the writing portions of the [`TcpStream`] should be shut down. |
| /// |
| Both, |
| } |
| |
| #[doc(hidden)] |
| trait NetInt { |
| fn from_be(i: Self) -> Self; |
| fn to_be(&self) -> Self; |
| } |
| macro_rules! doit { |
| ($($t:ident)*) => ($(impl NetInt for $t { |
| fn from_be(i: Self) -> Self { <$t>::from_be(i) } |
| fn to_be(&self) -> Self { <$t>::to_be(*self) } |
| })*) |
| } |
| doit! { i8 i16 i32 i64 isize u8 u16 u32 u64 usize } |
| |
| fn hton<I: NetInt>(i: I) -> I { i.to_be() } |
| fn ntoh<I: NetInt>(i: I) -> I { I::from_be(i) } |
| |
| fn each_addr<A: ToSocketAddrs, F, T>(addr: A, mut f: F) -> io::Result<T> |
| where F: FnMut(&SocketAddr) -> io::Result<T> |
| { |
| let mut last_err = None; |
| for addr in addr.to_socket_addrs()? { |
| match f(&addr) { |
| Ok(l) => return Ok(l), |
| Err(e) => last_err = Some(e), |
| } |
| } |
| Err(last_err.unwrap_or_else(|| { |
| Error::new(ErrorKind::InvalidInput, |
| "could not resolve to any addresses") |
| })) |
| } |