module Network.Wai.Handler.Warp.HTTP2.HPACK where
import Control.Arrow (first)
import qualified Control.Exception as E
import Data.ByteString.Builder (Builder)
import qualified Data.ByteString.Char8 as B8
import Data.CaseInsensitive (foldedCase)
import Data.IORef (readIORef, writeIORef)
import Network.HPACK
import qualified Network.HTTP.Types as H
import Network.HTTP2
import Network.Wai
import Network.Wai.Handler.Warp.HTTP2.Types
import Network.Wai.Handler.Warp.Header
import Network.Wai.Handler.Warp.Response
import qualified Network.Wai.Handler.Warp.Settings as S
import Network.Wai.Handler.Warp.Types
hpackEncodeHeader :: Context -> InternalInfo -> S.Settings -> Response
-> IO Builder
hpackEncodeHeader Context{encodeDynamicTable} ii settings rsp = do
hdr1 <- addServerAndDate hdr0
let hdr2 = (":status", status) : map (first foldedCase) hdr1
ehdrtbl <- readIORef encodeDynamicTable
(ehdrtbl', builder) <- encodeHeaderBuilder defaultEncodeStrategy ehdrtbl hdr2
writeIORef encodeDynamicTable ehdrtbl'
return builder
where
hdr0 = responseHeaders rsp
status = B8.pack $ show $ H.statusCode $ responseStatus rsp
dc = dateCacher ii
rspidxhdr = indexResponseHeader hdr0
defServer = S.settingsServerName settings
addServerAndDate = addDate dc rspidxhdr . addServer defServer rspidxhdr
hpackDecodeHeader :: HeaderBlockFragment -> Context -> IO HeaderList
hpackDecodeHeader hdrblk Context{decodeDynamicTable} = do
hdrtbl <- readIORef decodeDynamicTable
(hdrtbl', hdr) <- decodeHeader hdrtbl hdrblk `E.onException` cleanup
writeIORef decodeDynamicTable hdrtbl'
return hdr
where
cleanup = E.throwIO $ ConnectionError CompressionError "cannot decompress the header"