From 81e5b48e95c13989de3611e15a9cedee6c312573 Mon Sep 17 00:00:00 2001
From: Mohd Tarmizi <mohtar@users.noreply.github.com>
Date: Tue, 4 Aug 2015 07:19:58 +0800
Subject: [PATCH] Disallow when getting peer name causes an error

---
 src/lib.rs | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/src/lib.rs b/src/lib.rs
index 3704ee3..1d47c1d 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -572,7 +572,10 @@ fn run_transport<F>(mut handler: F, transport: &mut Transport) where F: FnMut(Re
             Err(e) => panic!(e.to_string()),
         };
         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,
         };
         if allow {