use multiaddr::{Multiaddr, Protocol};
pub fn address_translation(original: &Multiaddr, observed: &Multiaddr) -> Option<Multiaddr> {
original.replace(0, move |proto| match proto {
Protocol::Ip4(_)
| Protocol::Ip6(_)
| Protocol::Dns(_)
| Protocol::Dns4(_)
| Protocol::Dns6(_) => match observed.iter().next() {
x @ Some(Protocol::Ip4(_)) => x,
x @ Some(Protocol::Ip6(_)) => x,
x @ Some(Protocol::Dns(_)) => x,
x @ Some(Protocol::Dns4(_)) => x,
x @ Some(Protocol::Dns6(_)) => x,
_ => None,
},
_ => None,
})
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn test_address_translation() {
struct Test {
original: Multiaddr,
observed: Multiaddr,
expected: Multiaddr,
}
let tests = vec![
Test {
original: "/ip4/192.0.2.1/tcp/1".parse().unwrap(),
observed: "/ip4/192.0.2.2/tcp/2".parse().unwrap(),
expected: "/ip4/192.0.2.2/tcp/1".parse().unwrap(),
},
Test {
original: "/ip6/2001:db8:0:0:0:0:0:0/tcp/1".parse().unwrap(),
observed: "/ip6/2001:db8:0:0:0:0:0:1/tcp/2".parse().unwrap(),
expected: "/ip6/2001:db8:0:0:0:0:0:1/tcp/1".parse().unwrap(),
},
Test {
original: "/ip4/192.0.2.1/tcp/1".parse().unwrap(),
observed: "/ip6/2001:db8:0:0:0:0:0:1/tcp/2".parse().unwrap(),
expected: "/ip6/2001:db8:0:0:0:0:0:1/tcp/1".parse().unwrap(),
},
Test {
original: "/ip6/2001:db8:0:0:0:0:0:0/tcp/1".parse().unwrap(),
observed: "/ip4/192.0.2.2/tcp/2".parse().unwrap(),
expected: "/ip4/192.0.2.2/tcp/1".parse().unwrap(),
},
Test {
original: "/dns4/foo/tcp/1".parse().unwrap(),
observed: "/dns4/bar/tcp/2".parse().unwrap(),
expected: "/dns4/bar/tcp/1".parse().unwrap(),
},
Test {
original: "/ip4/192.0.2.1/tcp/1".parse().unwrap(),
observed: "/dns4/bar/tcp/2".parse().unwrap(),
expected: "/dns4/bar/tcp/1".parse().unwrap(),
},
];
for test in tests.iter() {
assert_eq!(
address_translation(&test.original, &test.observed),
Some(test.expected.clone())
);
}
}
}