Make extended caps mutable

1 files changed, 6 insertions, 1 deletions

diff --git a/src/Network/BitTorrent/Exchange/Message.hs b/src/Network/BitTorrent/Exchange/Message.hs
index b879e212..17ec7da6 100644
--- a/src/Network/BitTorrent/Exchange/Message.hs
+++ b/src/Network/BitTorrent/Exchange/Message.hs
@@ -62,6 +62,7 @@ module Network.BitTorrent.Exchange.Message
        , ExtendedExtension
        , ExtendedCaps      (..)
        , ExtendedHandshake (..)
+       , nullExtendedHandshake
        , ExtendedMetadata  (..)
        ) where
 
@@ -411,7 +412,7 @@ data ExtendedHandshake = ExtendedHandshake
   } deriving (Show, Eq, Typeable)
 
 instance Default ExtendedHandshake where
-  def = ExtendedHandshake Nothing Nothing def Nothing Nothing Nothing
+  def = nullExtendedHandshake def
 
 instance BEncode ExtendedHandshake where
   toBEncode ExtendedHandshake {..} = toDict $
@@ -439,6 +440,10 @@ instance Pretty ExtendedHandshake where
 instance PeerMessage ExtendedHandshake where
   envelop c = envelop c . EHandshake
 
+nullExtendedHandshake :: ExtendedCaps -> ExtendedHandshake
+nullExtendedHandshake caps
+  = ExtendedHandshake Nothing Nothing caps Nothing Nothing Nothing
+
 {-----------------------------------------------------------------------
 -- Metadata exchange
 -----------------------------------------------------------------------}