prevent panic from type assertions

tonobo · Aug 24, 2018 · 8910bb0 · 8910bb0
1 parent dd39cdc
commit 8910bb0
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/imcp/icmp.go b/imcp/icmp.go
@@ -133,9 +133,14 @@ func listenForSpecific(conn *icmp.PacketConn, deadline time.Time, neededPeer, ne
 			index := bytes.Index(body, sent[:4])
 			if index > 0 {
 				x, _ := icmp.ParseMessage(1, body[index:])
-				seq := x.Body.(*icmp.Echo).Seq
-				if seq == needSeq {
-					return peer.String(), "", nil
+				switch x.Body.(type) {
+				case *icmp.Echo:
+					seq := x.Body.(*icmp.Echo).Seq
+					if seq == needSeq {
+						return peer.String(), "", nil
+					}
+				default:
+					// ignore
 				}
 			}
 		}