Throw exception if query takes too long

1 files changed, 13 insertions, 1 deletions

diff --git a/src/Network/KRPC/Message.hs b/src/Network/KRPC/Message.hs
index 3bbfb1db..0bd34400 100644
--- a/src/Network/KRPC/Message.hs
+++ b/src/Network/KRPC/Message.hs
@@ -30,6 +30,7 @@ module Network.KRPC.Message
        , decodeError
        , unknownMethod
        , unknownMessage
+       , timeoutExpired
 
          -- * Query
        , KQuery(..)
@@ -130,17 +131,28 @@ instance BEncode KError where
 
 instance Exception KError
 
+-- | Happen when some handler fail.
 serverError :: SomeException -> TransactionId -> KError
 serverError e = KError ServerError (BC.pack (show e))
 
+-- | Received 'queryArgs' or 'respVals' can not be decoded.
 decodeError :: String -> TransactionId -> KError
 decodeError msg = KError ProtocolError (BC.pack msg)
 
+-- | If /remote/ node send query /this/ node doesn't know about then
+-- this error message should be sent in response.
 unknownMethod :: MethodName -> TransactionId -> KError
 unknownMethod = KError MethodUnknown
 
+-- | A remote node has send some 'KMessage' this node is unable to
+-- decode.
 unknownMessage :: String -> KError
-unknownMessage msg = KError ProtocolError (BC.pack msg) ""
+unknownMessage msg = KError ProtocolError (BC.pack msg) unknownTransaction
+
+-- | A /remote/ node is not responding to the /our/ request the for
+-- specified period of time.
+timeoutExpired :: TransactionId -> KError
+timeoutExpired = KError GenericError "timeout expired"
 
 {-----------------------------------------------------------------------
 -- Query messages