WIP: Mainline DHT rewrite.

1 files changed, 7 insertions, 0 deletions

diff --git a/src/Network/Address.hs b/src/Network/Address.hs
index 0fbd191a..d34b0431 100644
--- a/src/Network/Address.hs
+++ b/src/Network/Address.hs
@@ -83,6 +83,7 @@ module Network.Address
          -- * Utils
        , libUserAgent
        , sockAddrPort
+       , withPort
        , getBindAddress
        ) where
 
@@ -147,6 +148,12 @@ sockAddrPort (SockAddrInet6 p _ _ _) = Just p
 sockAddrPort _                       = Nothing
 {-# INLINE sockAddrPort #-}
 
+withPort :: SockAddr -> PortNumber -> SockAddr
+withPort (SockAddrInet  _ ip           ) port = SockAddrInet port ip
+withPort (SockAddrInet6 _ flow ip scope) port = SockAddrInet6 port flow ip scope
+withPort addr                            _    = addr
+{-# INLINE withPort #-}
+
 instance Address a => Address (NodeAddr a) where
   toSockAddr NodeAddr {..} = setPort nodePort $ toSockAddr nodeHost
   fromSockAddr sa = NodeAddr <$> fromSockAddr sa <*> sockAddrPort sa