Merge pull request #169 from wackbyte/arbitrary-for-socketaddr

Implement `Arbitrary` for `SocketAddr{,V4,V6}`

Author: Manishearth

src/lib.rs

@@ -47,7 +47,7 @@ use std::borrow::{Cow, ToOwned};
 use std::collections::{BTreeMap, BTreeSet, BinaryHeap, HashMap, HashSet, LinkedList, VecDeque};
 use std::ffi::{CString, OsString};
 use std::hash::BuildHasher;
-use std::net::{IpAddr, Ipv4Addr, Ipv6Addr};
+use std::net::{IpAddr, Ipv4Addr, Ipv6Addr, SocketAddr, SocketAddrV4, SocketAddrV6};
 use std::ops::Bound;
 use std::path::PathBuf;
 use std::rc::Rc;
@@ -1200,6 +1200,59 @@ impl<'a> Arbitrary<'a> for IpAddr {
     }
 }
 
+impl<'a> Arbitrary<'a> for SocketAddrV4 {
+    fn arbitrary(u: &mut Unstructured<'a>) -> Result<Self> {
+        Ok(SocketAddrV4::new(u.arbitrary()?, u.arbitrary()?))
+    }
+
+    #[inline]
+    fn size_hint(depth: usize) -> (usize, Option<usize>) {
+        size_hint::and(Ipv4Addr::size_hint(depth), u16::size_hint(depth))
+    }
+}
+
+impl<'a> Arbitrary<'a> for SocketAddrV6 {
+    fn arbitrary(u: &mut Unstructured<'a>) -> Result<Self> {
+        Ok(SocketAddrV6::new(
+            u.arbitrary()?,
+            u.arbitrary()?,
+            u.arbitrary()?,
+            u.arbitrary()?,
+        ))
+    }
+
+    #[inline]
+    fn size_hint(depth: usize) -> (usize, Option<usize>) {
+        size_hint::and(
+            Ipv6Addr::size_hint(depth),
+            size_hint::and(
+                u16::size_hint(depth),
+                size_hint::and(u32::size_hint(depth), u32::size_hint(depth)),
+            ),
+        )
+    }
+}
+
+impl<'a> Arbitrary<'a> for SocketAddr {
+    fn arbitrary(u: &mut Unstructured<'a>) -> Result<Self> {
+        if u.arbitrary()? {
+            Ok(SocketAddr::V4(u.arbitrary()?))
+        } else {
+            Ok(SocketAddr::V6(u.arbitrary()?))
+        }
+    }
+
+    fn size_hint(depth: usize) -> (usize, Option<usize>) {
+        size_hint::and(
+            bool::size_hint(depth),
+            size_hint::or(
+                SocketAddrV4::size_hint(depth),
+                SocketAddrV6::size_hint(depth),
+            ),
+        )
+    }
+}
+
 #[cfg(test)]
 mod test {
     use super::*;