Untested announce-peer command for bittorrent dht.

1 files changed, 4 insertions, 0 deletions

diff --git a/src/Network/BitTorrent/DHT/Token.hs b/src/Network/BitTorrent/DHT/Token.hs
index 756b5a98..3da59c53 100644
--- a/src/Network/BitTorrent/DHT/Token.hs
+++ b/src/Network/BitTorrent/DHT/Token.hs
@@ -44,6 +44,7 @@ module Network.BitTorrent.DHT.Token
        , Network.BitTorrent.DHT.Token.update
        ) where
 
+import Control.Arrow
 import Control.Monad.State
 #ifdef VERSION_bencoding
 import Data.BEncode (BEncode)
@@ -75,6 +76,9 @@ newtype Token = Token BS.ByteString
 instance Show Token where
     show (Token bs) = B8.unpack $ Base16.encode bs
 
+instance Read Token where
+    readsPrec i s = pure $ (Token *** B8.unpack) $ Base16.decode (B8.pack s)
+
 -- | Meaningless token, for testing purposes only.
 instance Default Token where
   def = makeToken (0::Int) 0