{-# LANGUAGE OverloadedStrings #-}

module PacketSpec where

import Data.ByteString (ByteString)
import qualified Data.ByteString as BS
import qualified Data.ByteString.Internal as BS
import Data.IORef
import Data.Tuple (swap)
import Network.UDP
import Test.Hspec

import Network.QUIC.Internal

import Config

spec :: Spec
spec = do
    serverConf <- runIO makeTestServerConfig
    describe "test vector" $ do
        it "describes example of Client Initial version 1" $ do
            conns <- makeConnections serverConf Version1
            checkBinary conns 2 clientInitialPacketBinaryV1
        it "describes example of Server Initial version 1" $ do
            conns <- swap <$> makeConnections serverConf Version1
            checkBinary conns 1 serverInitialPacketBinaryV1
        it "describes example of Client Initial version 2" $ do
            conns <- makeConnections serverConf Version2
            checkBinary conns 2 clientInitialPacketBinaryV2
        it "describes example of Server Initial version 2" $ do
            conns <- swap <$> makeConnections serverConf Version2
            checkBinary conns 1 serverInitialPacketBinaryV2

clientChosenCID :: CID
clientChosenCID = toCID $ dec16 "8394c8f03e515708"

makeConnections :: ServerConfig -> Version -> IO (Connection, Connection)
makeConnections conf v = do
    let noLog _ = return ()
    let serverCID = clientChosenCID
        clientCID = toCID ""
        serverAuthCIDs = defaultAuthCIDs { initSrcCID = Just serverCID
                                         , origDstCID = Just serverCID
                                         }
        clientAuthCIDs = defaultAuthCIDs { initSrcCID = Just clientCID }
        -- dummy
    let clientConf = testClientConfig
    us <- clientSocket "127.0.0.1" "2000" False
    q <- newRecvQ
    sref <- newIORef us
    let ver = v
        verInfo = VersionInfo ver [ver]
    ----
    clientConn <- clientConnection clientConf verInfo clientAuthCIDs serverAuthCIDs noLog noLog defaultHooks sref q undefined undefined
    initializeCoder clientConn InitialLevel $ initialSecrets ver serverCID
    serverConn <- serverConnection conf verInfo serverAuthCIDs clientAuthCIDs noLog noLog defaultHooks sref q undefined undefined
    initializeCoder serverConn InitialLevel $ initialSecrets ver serverCID
    ----
    return (clientConn, serverConn)

checkBinary :: (Connection,Connection)-> PacketNumber -> ByteString -> IO ()
checkBinary (senderConn,recverConn) pn bin = do
    ---- Decoding by the receiver
    (PacketIC (CryptPacket header crypt) lvl _, _) <- decodePacket bin
    ---- Cecrypting by the receiver
    Just plain <- decryptCrypt recverConn crypt lvl
    ---- Checking
    plainPacketNumber plain `shouldBe` pn
    ---- Encoding by the sender
    let ppkt = PlainPacket header plain
    bin' <- BS.createAndTrim 4096 $ \buf ->
        fst <$> encodePlainPacket senderConn (SizedBuffer buf 2048) ppkt Nothing
    ---- Decoding by the receiver again
    (PacketIC (CryptPacket header' crypt') lvl' _, _) <- decodePacket bin'
    ---- Cecrypting by the receiver again
    Just plain' <- decryptCrypt recverConn crypt' lvl'
    ---- Checking
    header' `shouldBe` header
    plainFrames plain' `shouldBe` plainFrames plain

clientInitialPacketBinaryV1 :: ByteString
clientInitialPacketBinaryV1 = dec16 $ BS.concat [
    "c000000001088394c8f03e5157080000449e7b9aec34d1b1c98dd7689fb8ec11"
  , "d242b123dc9bd8bab936b47d92ec356c0bab7df5976d27cd449f63300099f399"
  , "1c260ec4c60d17b31f8429157bb35a1282a643a8d2262cad67500cadb8e7378c"
  , "8eb7539ec4d4905fed1bee1fc8aafba17c750e2c7ace01e6005f80fcb7df6212"
  , "30c83711b39343fa028cea7f7fb5ff89eac2308249a02252155e2347b63d58c5"
  , "457afd84d05dfffdb20392844ae812154682e9cf012f9021a6f0be17ddd0c208"
  , "4dce25ff9b06cde535d0f920a2db1bf362c23e596d11a4f5a6cf3948838a3aec"
  , "4e15daf8500a6ef69ec4e3feb6b1d98e610ac8b7ec3faf6ad760b7bad1db4ba3"
  , "485e8a94dc250ae3fdb41ed15fb6a8e5eba0fc3dd60bc8e30c5c4287e53805db"
  , "059ae0648db2f64264ed5e39be2e20d82df566da8dd5998ccabdae053060ae6c"
  , "7b4378e846d29f37ed7b4ea9ec5d82e7961b7f25a9323851f681d582363aa5f8"
  , "9937f5a67258bf63ad6f1a0b1d96dbd4faddfcefc5266ba6611722395c906556"
  , "be52afe3f565636ad1b17d508b73d8743eeb524be22b3dcbc2c7468d54119c74"
  , "68449a13d8e3b95811a198f3491de3e7fe942b330407abf82a4ed7c1b311663a"
  , "c69890f4157015853d91e923037c227a33cdd5ec281ca3f79c44546b9d90ca00"
  , "f064c99e3dd97911d39fe9c5d0b23a229a234cb36186c4819e8b9c5927726632"
  , "291d6a418211cc2962e20fe47feb3edf330f2c603a9d48c0fcb5699dbfe58964"
  , "25c5bac4aee82e57a85aaf4e2513e4f05796b07ba2ee47d80506f8d2c25e50fd"
  , "14de71e6c418559302f939b0e1abd576f279c4b2e0feb85c1f28ff18f58891ff"
  , "ef132eef2fa09346aee33c28eb130ff28f5b766953334113211996d20011a198"
  , "e3fc433f9f2541010ae17c1bf202580f6047472fb36857fe843b19f5984009dd"
  , "c324044e847a4f4a0ab34f719595de37252d6235365e9b84392b061085349d73"
  , "203a4a13e96f5432ec0fd4a1ee65accdd5e3904df54c1da510b0ff20dcc0c77f"
  , "cb2c0e0eb605cb0504db87632cf3d8b4dae6e705769d1de354270123cb11450e"
  , "fc60ac47683d7b8d0f811365565fd98c4c8eb936bcab8d069fc33bd801b03ade"
  , "a2e1fbc5aa463d08ca19896d2bf59a071b851e6c239052172f296bfb5e724047"
  , "90a2181014f3b94a4e97d117b438130368cc39dbb2d198065ae3986547926cd2"
  , "162f40a29f0c3c8745c0f50fba3852e566d44575c29d39a03f0cda721984b6f4"
  , "40591f355e12d439ff150aab7613499dbd49adabc8676eef023b15b65bfc5ca0"
  , "6948109f23f350db82123535eb8a7433bdabcb909271a6ecbcb58b936a88cd4e"
  , "8f2e6ff5800175f113253d8fa9ca8885c2f552e657dc603f252e1a8e308f76f0"
  , "be79e2fb8f5d5fbbe2e30ecadd220723c8c0aea8078cdfcb3868263ff8f09400"
  , "54da48781893a7e49ad5aff4af300cd804a6b6279ab3ff3afb64491c85194aab"
  , "760d58a606654f9f4400e8b38591356fbf6425aca26dc85244259ff2b19c41b9"
  , "f96f3ca9ec1dde434da7d2d392b905ddf3d1f9af93d1af5950bd493f5aa731b4"
  , "056df31bd267b6b90a079831aaf579be0a39013137aac6d404f518cfd4684064"
  , "7e78bfe706ca4cf5e9c5453e9f7cfd2b8b4c8d169a44e55c88d4a9a7f9474241"
  , "e221af44860018ab0856972e194cd934"
  ]

clientInitialPacketBinaryV2 :: ByteString
clientInitialPacketBinaryV2 = dec16 $ BS.concat [
    "d76b3343cf088394c8f03e5157080000449ea0c95e82ffe67b6abcdb4298b485"
  , "dd04de806071bf03dceebfa162e75d6c96058bdbfb127cdfcbf903388e99ad04"
  , "9f9a3dd4425ae4d0992cfff18ecf0fdb5a842d09747052f17ac2053d21f57c5d"
  , "250f2c4f0e0202b70785b7946e992e58a59ac52dea6774d4f03b55545243cf1a"
  , "12834e3f249a78d395e0d18f4d766004f1a2674802a747eaa901c3f10cda5500"
  , "cb9122faa9f1df66c392079a1b40f0de1c6054196a11cbea40afb6ef5253cd68"
  , "18f6625efce3b6def6ba7e4b37a40f7732e093daa7d52190935b8da58976ff33"
  , "12ae50b187c1433c0f028edcc4c2838b6a9bfc226ca4b4530e7a4ccee1bfa2a3"
  , "d396ae5a3fb512384b2fdd851f784a65e03f2c4fbe11a53c7777c023462239dd"
  , "6f7521a3f6c7d5dd3ec9b3f233773d4b46d23cc375eb198c63301c21801f6520"
  , "bcfb7966fc49b393f0061d974a2706df8c4a9449f11d7f3d2dcbb90c6b877045"
  , "636e7c0c0fe4eb0f697545460c806910d2c355f1d253bc9d2452aaa549e27a1f"
  , "ac7cf4ed77f322e8fa894b6a83810a34b361901751a6f5eb65a0326e07de7c12"
  , "16ccce2d0193f958bb3850a833f7ae432b65bc5a53975c155aa4bcb4f7b2c4e5"
  , "4df16efaf6ddea94e2c50b4cd1dfe06017e0e9d02900cffe1935e0491d77ffb4"
  , "fdf85290fdd893d577b1131a610ef6a5c32b2ee0293617a37cbb08b847741c3b"
  , "8017c25ca9052ca1079d8b78aebd47876d330a30f6a8c6d61dd1ab5589329de7"
  , "14d19d61370f8149748c72f132f0fc99f34d766c6938597040d8f9e2bb522ff9"
  , "9c63a344d6a2ae8aa8e51b7b90a4a806105fcbca31506c446151adfeceb51b91"
  , "abfe43960977c87471cf9ad4074d30e10d6a7f03c63bd5d4317f68ff325ba3bd"
  , "80bf4dc8b52a0ba031758022eb025cdd770b44d6d6cf0670f4e990b22347a7db"
  , "848265e3e5eb72dfe8299ad7481a408322cac55786e52f633b2fb6b614eaed18"
  , "d703dd84045a274ae8bfa73379661388d6991fe39b0d93debb41700b41f90a15"
  , "c4d526250235ddcd6776fc77bc97e7a417ebcb31600d01e57f32162a8560cacc"
  , "7e27a096d37a1a86952ec71bd89a3e9a30a2a26162984d7740f81193e8238e61"
  , "f6b5b984d4d3dfa033c1bb7e4f0037febf406d91c0dccf32acf423cfa1e70710"
  , "10d3f270121b493ce85054ef58bada42310138fe081adb04e2bd901f2f13458b"
  , "3d6758158197107c14ebb193230cd1157380aa79cae1374a7c1e5bbcb80ee23e"
  , "06ebfde206bfb0fcbc0edc4ebec309661bdd908d532eb0c6adc38b7ca7331dce"
  , "8dfce39ab71e7c32d318d136b6100671a1ae6a6600e3899f31f0eed19e3417d1"
  , "34b90c9058f8632c798d4490da4987307cba922d61c39805d072b589bd52fdf1"
  , "e86215c2d54e6670e07383a27bbffb5addf47d66aa85a0c6f9f32e59d85a44dd"
  , "5d3b22dc2be80919b490437ae4f36a0ae55edf1d0b5cb4e9a3ecabee93dfc6e3"
  , "8d209d0fa6536d27a5d6fbb17641cde27525d61093f1b28072d111b2b4ae5f89"
  , "d5974ee12e5cf7d5da4d6a31123041f33e61407e76cffcdcfd7e19ba58cf4b53"
  , "6f4c4938ae79324dc402894b44faf8afbab35282ab659d13c93f70412e85cb19"
  , "9a37ddec600545473cfb5a05e08d0b209973b2172b4d21fb69745a262ccde96b"
  , "a18b2faa745b6fe189cf772a9f84cbfc"
  ]

serverInitialPacketBinaryV1 :: ByteString
serverInitialPacketBinaryV1 = dec16 $ BS.concat [
    "cf000000010008f067a5502a4262b5004075c0d95a482cd0991cd25b0aac406a"
  , "5816b6394100f37a1c69797554780bb38cc5a99f5ede4cf73c3ec2493a1839b3"
  , "dbcba3f6ea46c5b7684df3548e7ddeb9c3bf9c73cc3f3bded74b562bfb19fb84"
  , "022f8ef4cdd93795d77d06edbb7aaf2f58891850abbdca3d20398c276456cbc4"
  , "2158407dd074ee"
  ]

serverInitialPacketBinaryV2 :: ByteString
serverInitialPacketBinaryV2 = dec16 $ BS.concat [
    "dc6b3343cf0008f067a5502a4262b5004075d92faaf16f05d8a4398c47089698"
  , "baeea26b91eb761d9b89237bbf87263017915358230035f7fd3945d88965cf17"
  , "f9af6e16886c61bfc703106fbaf3cb4cfa52382dd16a393e42757507698075b2"
  , "c984c707f0a0812d8cd5a6881eaf21ceda98f4bd23f6fe1a3e2c43edd9ce7ca8"
  , "4bed8521e2e140"
  ]