Disallow when getting peer name causes an error

This commit is contained in:
Mohd Tarmizi 2015-08-04 07:19:58 +08:00
parent 15d3aed9f2
commit 81e5b48e95
1 changed files with 4 additions and 1 deletions

View File

@ -572,7 +572,10 @@ fn run_transport<F>(mut handler: F, transport: &mut Transport) where F: FnMut(Re
Err(e) => panic!(e.to_string()), Err(e) => panic!(e.to_string()),
}; };
let allow = match addrs { let allow = match addrs {
Some(ref addrs) => addrs.contains(&sock.peer().unwrap()), Some(ref addrs) => match sock.peer() {
Ok(ref addr) => addrs.contains(addr),
Err(_) => false,
},
None => true, None => true,
}; };
if allow { if allow {