grassrootseconomics/openethereum
5
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

ip_utils tests

Browse Source
This commit is contained in:
arkpar 2016-02-16 17:55:37 +01:00
parent a4ea0737b2
commit f4fa747cd0
1 changed files with 28 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
util/src/network/ip_utils.rs
View File

@ -43,23 +43,19 @@ mod getinterfaces {
let sa: *const sockaddr_in = unsafe { mem::transmute(sa) }; let sa: *const sockaddr_in = unsafe { mem::transmute(sa) };
let sa = & unsafe { *sa }; let sa = & unsafe { *sa };
let (addr, port) = (sa.sin_addr.s_addr, sa.sin_port); let (addr, port) = (sa.sin_addr.s_addr, sa.sin_port);
( (IpAddr::V4(Ipv4Addr::new(
IpAddr::V4(Ipv4Addr::new(
(addr & 0x000000FF) as u8, (addr & 0x000000FF) as u8,
((addr & 0x0000FF00) >> 8) as u8, ((addr & 0x0000FF00) >> 8) as u8,
((addr & 0x00FF0000) >> 16) as u8, ((addr & 0x00FF0000) >> 16) as u8,
((addr & 0xFF000000) >> 24) as u8, ((addr & 0xFF000000) >> 24) as u8)),
)), port)
port
)
}, },
AF_INET6 => { AF_INET6 => {
let sa: *const sockaddr_in6 = unsafe { mem::transmute(sa) }; let sa: *const sockaddr_in6 = unsafe { mem::transmute(sa) };
let sa = & unsafe { *sa }; let sa = & unsafe { *sa };
let (addr, port) = (sa.sin6_addr.s6_addr, sa.sin6_port); let (addr, port) = (sa.sin6_addr.s6_addr, sa.sin6_port);
let addr: [u16; 8] = unsafe { mem::transmute(addr) }; let addr: [u16; 8] = unsafe { mem::transmute(addr) };
( (IpAddr::V6(Ipv6Addr::new(
IpAddr::V6(Ipv6Addr::new(
addr[0], addr[0],
addr[1], addr[1],
addr[2], addr[2],
@ -67,10 +63,8 @@ mod getinterfaces {
addr[4], addr[4],
addr[5], addr[5],
addr[6], addr[6],
addr[7], addr[7])),
)), port)
port
)
}, },
_ => return None, _ => return None,
}; };
@ -174,3 +168,15 @@ pub fn map_external_address(local: &NodeEndpoint) -> Option<NodeEndpoint> {
None None
} }
#[test]
fn can_select_public_address() {
let pub_address = select_public_address(40477);
assert!(pub_address.port() == 40477);
}
#[test]
fn can_map_external_address_or_fail() {
let pub_address = select_public_address(40478);
let _ = map_external_address(&NodeEndpoint { address: pub_address, udp_port: 40478 });
}