Skip to content

Commit

Permalink
Migrated io::net::udp over to ToSocketAddr
Browse files Browse the repository at this point in the history
UdpSocket constructor methods now use ToSocketAddr trait instead of
SocketAddr.

[breaking-change]
  • Loading branch information
netvl committed Nov 5, 2014
1 parent ac84674 commit 7e3344b
Show file tree
Hide file tree
Showing 3 changed files with 34 additions and 20 deletions.
25 changes: 22 additions & 3 deletions src/libstd/io/net/mod.rs
Expand Up @@ -10,11 +10,11 @@

//! Networking I/O

use io::{IoError, InvalidInput};
use io::{IoError, IoResult, InvalidInput};
use option::None;
use result::{Result, Ok, Err};
use rt::rtio;
use self::ip::{Ipv4Addr, Ipv6Addr, IpAddr, ToSocketAddr};
use self::ip::{Ipv4Addr, Ipv6Addr, IpAddr, SocketAddr, ToSocketAddr};

pub use self::addrinfo::get_host_addresses;

Expand Down Expand Up @@ -42,7 +42,7 @@ fn from_rtio(ip: rtio::IpAddr) -> IpAddr {
}
}

fn with_addresses<A: ToSocketAddr, T>(
fn with_addresses_io<A: ToSocketAddr, T>(
addr: A,
action: |&mut rtio::IoFactory, rtio::SocketAddr| -> Result<T, rtio::IoError>
) -> Result<T, IoError> {
Expand All @@ -63,3 +63,22 @@ fn with_addresses<A: ToSocketAddr, T>(
}
Err(err)
}

fn with_addresses<A: ToSocketAddr, T>(addr: A, action: |SocketAddr| -> IoResult<T>)
-> IoResult<T> {
const DEFAULT_ERROR: IoError = IoError {
kind: InvalidInput,
desc: "no addresses found for hostname",
detail: None
};

let addresses = try!(addr.to_socket_addr_all());
let mut err = DEFAULT_ERROR;
for addr in addresses.into_iter() {
match action(addr) {
Ok(r) => return Ok(r),
Err(e) => err = e
}
}
Err(err)
}
6 changes: 3 additions & 3 deletions src/libstd/io/net/tcp.rs
Expand Up @@ -67,7 +67,7 @@ impl TcpStream {
/// trait can be supplied for the address; see this trait documentation for
/// concrete examples.
pub fn connect<A: ToSocketAddr>(addr: A) -> IoResult<TcpStream> {
super::with_addresses(addr, |io, addr| io.tcp_connect(addr, None).map(TcpStream::new))
super::with_addresses_io(addr, |io, addr| io.tcp_connect(addr, None).map(TcpStream::new))
}

/// Creates a TCP connection to a remote socket address, timing out after
Expand All @@ -89,7 +89,7 @@ impl TcpStream {
return Err(standard_error(TimedOut));
}

super::with_addresses(addr, |io, addr|
super::with_addresses_io(addr, |io, addr|
io.tcp_connect(addr, Some(timeout.num_milliseconds() as u64)).map(TcpStream::new)
)
}
Expand Down Expand Up @@ -324,7 +324,7 @@ impl TcpListener {
/// to this listener. The port allocated can be queried via the
/// `socket_name` function.
pub fn bind<A: ToSocketAddr>(addr: A) -> IoResult<TcpListener> {
super::with_addresses(addr, |io, addr| io.tcp_bind(addr).map(|l| TcpListener { obj: l }))
super::with_addresses_io(addr, |io, addr| io.tcp_bind(addr).map(|l| TcpListener { obj: l }))
}

/// Returns the local socket address of this listener.
Expand Down
23 changes: 9 additions & 14 deletions src/libstd/io/net/udp.rs
Expand Up @@ -16,7 +16,7 @@
//! datagram protocol.

use clone::Clone;
use io::net::ip::{SocketAddr, IpAddr};
use io::net::ip::{SocketAddr, IpAddr, ToSocketAddr};
use io::{Reader, Writer, IoResult, IoError};
use kinds::Send;
use boxed::Box;
Expand Down Expand Up @@ -65,18 +65,13 @@ pub struct UdpSocket {

impl UdpSocket {
/// Creates a UDP socket from the given socket address.
pub fn bind(addr: SocketAddr) -> IoResult<UdpSocket> {
let SocketAddr { ip, port } = addr;
LocalIo::maybe_raise(|io| {
let addr = rtio::SocketAddr { ip: super::to_rtio(ip), port: port };
io.udp_bind(addr).map(|s| UdpSocket { obj: s })
}).map_err(IoError::from_rtio_error)
pub fn bind<A: ToSocketAddr>(addr: A) -> IoResult<UdpSocket> {
super::with_addresses_io(addr, |io, addr| io.udp_bind(addr).map(|s| UdpSocket { obj: s }))
}

/// Receives data from the socket. On success, returns the number of bytes
/// read and the address from whence the data came.
pub fn recv_from(&mut self, buf: &mut [u8])
-> IoResult<(uint, SocketAddr)> {
pub fn recv_from(&mut self, buf: &mut [u8]) -> IoResult<(uint, SocketAddr)> {
match self.obj.recv_from(buf) {
Ok((amt, rtio::SocketAddr { ip, port })) => {
Ok((amt, SocketAddr { ip: super::from_rtio(ip), port: port }))
Expand All @@ -87,11 +82,11 @@ impl UdpSocket {

/// Sends data on the socket to the given address. Returns nothing on
/// success.
pub fn send_to(&mut self, buf: &[u8], dst: SocketAddr) -> IoResult<()> {
self.obj.send_to(buf, rtio::SocketAddr {
ip: super::to_rtio(dst.ip),
port: dst.port,
}).map_err(IoError::from_rtio_error)
pub fn send_to<A: ToSocketAddr>(&mut self, buf: &[u8], addr: A) -> IoResult<()> {
super::with_addresses(addr, |addr| self.obj.send_to(buf, rtio::SocketAddr {
ip: super::to_rtio(addr.ip),
port: addr.port,
}).map_err(IoError::from_rtio_error))
}

/// Creates a `UdpStream`, which allows use of the `Reader` and `Writer`
Expand Down

0 comments on commit 7e3344b

Please sign in to comment.