{-# LANGUAGE DeriveGeneric #-}
{-# LANGUAGE RecordWildCards #-}

module Network.QUIC.Types.Resumption where

import Codec.Serialise
import GHC.Generics
import Network.TLS hiding (Version)
import Network.TLS.QUIC

import Network.QUIC.Imports
import Network.QUIC.Types.Frame
import Network.QUIC.Types.Packet

type SessionEstablish = SessionID -> SessionData -> IO (Maybe Ticket)

-- | Information about resumption
data ResumptionInfo = ResumptionInfo
    { ResumptionInfo -> Version
resumptionVersion :: Version
    , ResumptionInfo -> Maybe (Token, SessionData)
resumptionSession :: Maybe (SessionID, SessionData)
    , ResumptionInfo -> Token
resumptionToken :: Token
    , ResumptionInfo -> Bool
resumptionRetry :: Bool
    }
    deriving (ResumptionInfo -> ResumptionInfo -> Bool
(ResumptionInfo -> ResumptionInfo -> Bool)
-> (ResumptionInfo -> ResumptionInfo -> Bool) -> Eq ResumptionInfo
forall a. (a -> a -> Bool) -> (a -> a -> Bool) -> Eq a
$c== :: ResumptionInfo -> ResumptionInfo -> Bool
== :: ResumptionInfo -> ResumptionInfo -> Bool
$c/= :: ResumptionInfo -> ResumptionInfo -> Bool
/= :: ResumptionInfo -> ResumptionInfo -> Bool
Eq, Int -> ResumptionInfo -> ShowS
[ResumptionInfo] -> ShowS
ResumptionInfo -> String
(Int -> ResumptionInfo -> ShowS)
-> (ResumptionInfo -> String)
-> ([ResumptionInfo] -> ShowS)
-> Show ResumptionInfo
forall a.
(Int -> a -> ShowS) -> (a -> String) -> ([a] -> ShowS) -> Show a
$cshowsPrec :: Int -> ResumptionInfo -> ShowS
showsPrec :: Int -> ResumptionInfo -> ShowS
$cshow :: ResumptionInfo -> String
show :: ResumptionInfo -> String
$cshowList :: [ResumptionInfo] -> ShowS
showList :: [ResumptionInfo] -> ShowS
Show, (forall x. ResumptionInfo -> Rep ResumptionInfo x)
-> (forall x. Rep ResumptionInfo x -> ResumptionInfo)
-> Generic ResumptionInfo
forall x. Rep ResumptionInfo x -> ResumptionInfo
forall x. ResumptionInfo -> Rep ResumptionInfo x
forall a.
(forall x. a -> Rep a x) -> (forall x. Rep a x -> a) -> Generic a
$cfrom :: forall x. ResumptionInfo -> Rep ResumptionInfo x
from :: forall x. ResumptionInfo -> Rep ResumptionInfo x
$cto :: forall x. Rep ResumptionInfo x -> ResumptionInfo
to :: forall x. Rep ResumptionInfo x -> ResumptionInfo
Generic)

instance Serialise ResumptionInfo

defaultResumptionInfo :: ResumptionInfo
defaultResumptionInfo :: ResumptionInfo
defaultResumptionInfo =
    ResumptionInfo
        { resumptionVersion :: Version
resumptionVersion = Version
Version1
        , resumptionSession :: Maybe (Token, SessionData)
resumptionSession = Maybe (Token, SessionData)
forall a. Maybe a
Nothing
        , resumptionToken :: Token
resumptionToken = Token
emptyToken
        , resumptionRetry :: Bool
resumptionRetry = Bool
False
        }

-- | Is 0RTT possible?
is0RTTPossible :: ResumptionInfo -> Bool
is0RTTPossible :: ResumptionInfo -> Bool
is0RTTPossible ResumptionInfo{Bool
Maybe (Token, SessionData)
Token
Version
resumptionVersion :: ResumptionInfo -> Version
resumptionSession :: ResumptionInfo -> Maybe (Token, SessionData)
resumptionToken :: ResumptionInfo -> Token
resumptionRetry :: ResumptionInfo -> Bool
resumptionVersion :: Version
resumptionSession :: Maybe (Token, SessionData)
resumptionToken :: Token
resumptionRetry :: Bool
..} =
    Bool
rtt0OK Bool -> Bool -> Bool
&& (Bool -> Bool
not Bool
resumptionRetry Bool -> Bool -> Bool
|| Token
resumptionToken Token -> Token -> Bool
forall a. Eq a => a -> a -> Bool
/= Token
emptyToken)
  where
    rtt0OK :: Bool
rtt0OK = case Maybe (Token, SessionData)
resumptionSession of
        Maybe (Token, SessionData)
Nothing -> Bool
False
        Just (Token
_, SessionData
sd) -> SessionData -> Int
sessionMaxEarlyDataSize SessionData
sd Int -> Int -> Bool
forall a. Eq a => a -> a -> Bool
== Int
quicMaxEarlyDataSize

-- | Is resumption possible?
isResumptionPossible :: ResumptionInfo -> Bool
isResumptionPossible :: ResumptionInfo -> Bool
isResumptionPossible ResumptionInfo{Bool
Maybe (Token, SessionData)
Token
Version
resumptionVersion :: ResumptionInfo -> Version
resumptionSession :: ResumptionInfo -> Maybe (Token, SessionData)
resumptionToken :: ResumptionInfo -> Token
resumptionRetry :: ResumptionInfo -> Bool
resumptionVersion :: Version
resumptionSession :: Maybe (Token, SessionData)
resumptionToken :: Token
resumptionRetry :: Bool
..} = Maybe (Token, SessionData) -> Bool
forall a. Maybe a -> Bool
isJust Maybe (Token, SessionData)
resumptionSession

get0RTTCipher :: ResumptionInfo -> Maybe CipherID
get0RTTCipher :: ResumptionInfo -> Maybe CipherID
get0RTTCipher ResumptionInfo
ri = case ResumptionInfo -> Maybe (Token, SessionData)
resumptionSession ResumptionInfo
ri of
    Maybe (Token, SessionData)
Nothing -> Maybe CipherID
forall a. Maybe a
Nothing
    Just (Token
_, SessionData
sd) -> CipherID -> Maybe CipherID
forall a. a -> Maybe a
Just (CipherID -> Maybe CipherID) -> CipherID -> Maybe CipherID
forall a b. (a -> b) -> a -> b
$ SessionData -> CipherID
sessionCipher SessionData
sd