Add hashable instances for all address types

1 files changed, 15 insertions, 0 deletions

diff --git a/src/Network/BitTorrent/Core/PeerAddr.hs b/src/Network/BitTorrent/Core/PeerAddr.hs
index 4905f910..d634716c 100644
--- a/src/Network/BitTorrent/Core/PeerAddr.hs
+++ b/src/Network/BitTorrent/Core/PeerAddr.hs
@@ -176,6 +176,17 @@ instance Pretty IP where
   pretty = PP.text . show
   {-# INLINE pretty #-}
 
+instance Hashable IPv4 where
+  hashWithSalt = hashUsing toHostAddress
+  {-# INLINE hashWithSalt #-}
+
+instance Hashable IPv6 where
+  hashWithSalt s a = hashWithSalt s (toHostAddress6 a)
+
+instance Hashable IP where
+  hashWithSalt s (IPv4 h) = hashWithSalt s h
+  hashWithSalt s (IPv6 h) = hashWithSalt s h
+
 {-----------------------------------------------------------------------
 --  Peer addr
 -----------------------------------------------------------------------}
@@ -267,6 +278,10 @@ instance Pretty a => Pretty (PeerAddr a) where
     where
       paddr = pretty peerHost <> ":" <> text (show peerPort)
 
+instance Hashable a => Hashable (PeerAddr a) where
+  hashWithSalt s PeerAddr {..} =
+    s `hashWithSalt` peerId `hashWithSalt` peerHost `hashWithSalt` peerPort
+
 -- | Ports typically reserved for bittorrent P2P listener.
 defaultPorts :: [PortNumber]
 defaultPorts =  [6881..6889]