You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
132 lines
5.1 KiB
132 lines
5.1 KiB
{-# OPTIONS_HADDOCK hide #-} |
|
{-# LANGUAGE OverloadedStrings #-} |
|
|
|
module Network.Xmpp.Sasl.Mechanisms.DigestMd5 |
|
( digestMd5 |
|
) where |
|
|
|
import Control.Applicative |
|
import Control.Arrow (left) |
|
import Control.Monad |
|
import Control.Monad.Error |
|
import Control.Monad.State.Strict |
|
import Data.Maybe (fromJust, isJust) |
|
|
|
import qualified Crypto.Classes as CC |
|
|
|
import qualified Data.Binary as Binary |
|
import qualified Data.ByteString.Base64 as B64 |
|
import qualified Data.ByteString.Char8 as BS8 |
|
import qualified Data.ByteString.Lazy as BL |
|
import qualified Data.Digest.Pure.MD5 as MD5 |
|
import qualified Data.List as L |
|
|
|
import qualified Data.Text as Text |
|
import Data.Text (Text) |
|
import qualified Data.Text.Encoding as Text |
|
|
|
import Data.XML.Pickle |
|
|
|
import qualified Data.ByteString as BS |
|
|
|
import Data.XML.Types |
|
|
|
import Network.Xmpp.Stream |
|
import Network.Xmpp.Types |
|
import Network.Xmpp.Sasl.Common |
|
import Network.Xmpp.Sasl.StringPrep |
|
import Network.Xmpp.Sasl.Types |
|
|
|
|
|
|
|
xmppDigestMd5 :: Text -- ^ Authentication identity (authzid or username) |
|
-> Maybe Text -- ^ Authorization identity (authcid) |
|
-> Text -- ^ Password (authzid) |
|
-> ErrorT AuthFailure (StateT Stream IO) () |
|
xmppDigestMd5 authcid authzid password = do |
|
(ac, az, pw) <- prepCredentials authcid authzid password |
|
hn <- gets streamHostname |
|
xmppDigestMd5' (fromJust hn) ac az pw |
|
where |
|
xmppDigestMd5' :: Text -> Text -> Maybe Text -> Text -> ErrorT AuthFailure (StateT Stream IO) () |
|
xmppDigestMd5' hostname authcid authzid password = do |
|
-- Push element and receive the challenge. |
|
_ <- saslInit "DIGEST-MD5" Nothing -- TODO: Check boolean? |
|
pairs <- toPairs =<< saslFromJust =<< pullChallenge |
|
cnonce <- liftIO $ makeNonce |
|
_b <- respond . Just $ createResponse hostname pairs cnonce |
|
challenge2 <- pullFinalMessage |
|
return () |
|
where |
|
-- Produce the response to the challenge. |
|
createResponse :: Text |
|
-> Pairs |
|
-> BS.ByteString -- nonce |
|
-> BS.ByteString |
|
createResponse hostname pairs cnonce = let |
|
Just qop = L.lookup "qop" pairs -- TODO: proper handling |
|
Just nonce = L.lookup "nonce" pairs |
|
uname_ = Text.encodeUtf8 authcid |
|
passwd_ = Text.encodeUtf8 password |
|
-- Using Int instead of Word8 for random 1.0.0.0 (GHC 7) |
|
-- compatibility. |
|
|
|
nc = "00000001" |
|
digestURI = "xmpp/" `BS.append` Text.encodeUtf8 hostname |
|
digest = md5Digest |
|
uname_ |
|
(lookup "realm" pairs) |
|
passwd_ |
|
digestURI |
|
nc |
|
qop |
|
nonce |
|
cnonce |
|
response = BS.intercalate "," . map (BS.intercalate "=") $ |
|
[["username", quote uname_]] ++ |
|
case L.lookup "realm" pairs of |
|
Just realm -> [["realm" , quote realm ]] |
|
Nothing -> [] ++ |
|
[ ["nonce" , quote nonce ] |
|
, ["cnonce" , quote cnonce ] |
|
, ["nc" , nc ] |
|
, ["qop" , qop ] |
|
, ["digest-uri", quote digestURI] |
|
, ["response" , digest ] |
|
, ["charset" , "utf-8" ] |
|
] |
|
in B64.encode response |
|
hash :: [BS8.ByteString] -> BS8.ByteString |
|
hash = BS8.pack . show |
|
. (CC.hash' :: BS.ByteString -> MD5.MD5Digest) |
|
. BS.intercalate (":") |
|
hashRaw :: [BS8.ByteString] -> BS8.ByteString |
|
hashRaw = toStrict . Binary.encode . |
|
(CC.hash' :: BS.ByteString -> MD5.MD5Digest) . BS.intercalate (":") |
|
toStrict :: BL.ByteString -> BS8.ByteString |
|
toStrict = BS.concat . BL.toChunks |
|
-- TODO: this only handles MD5-sess |
|
md5Digest :: BS8.ByteString |
|
-> Maybe BS8.ByteString |
|
-> BS8.ByteString |
|
-> BS8.ByteString |
|
-> BS8.ByteString |
|
-> BS8.ByteString |
|
-> BS8.ByteString |
|
-> BS8.ByteString |
|
-> BS8.ByteString |
|
md5Digest uname realm password digestURI nc qop nonce cnonce = |
|
let ha1 = hash [ hashRaw [uname, maybe "" id realm, password] |
|
, nonce |
|
, cnonce |
|
] |
|
ha2 = hash ["AUTHENTICATE", digestURI] |
|
in hash [ha1, nonce, nc, cnonce, qop, ha2] |
|
|
|
digestMd5 :: Text -- ^ Authentication identity (authcid or username) |
|
-> Maybe Text -- ^ Authorization identity (authzid) |
|
-> Text -- ^ Password |
|
-> SaslHandler |
|
digestMd5 authcid authzid password = ( "DIGEST-MD5" |
|
, xmppDigestMd5 authcid authzid password |
|
)
|
|
|