theta/test/Test/Packets/L3.hs

module Test.Packets.L3
  (
    tests
  ) where

import           Data.Binary         (encode)
import qualified Data.ByteString     as B
import           Data.Word           (Word8 (..))
import           Hedgehog            (MonadGen (..), forAll, property, (===))
import           Hedgehog.Gen
import qualified Hedgehog.Range      as Range
import           Packets.L3
import           Test.Tasty
import           Test.Tasty.Hedgehog
import           Test.Tasty.HUnit

tests :: TestTree
tests = testGroup "L3" [ unitTests]

unitTests :: TestTree
unitTests = testGroup "Unit tests" [ testIpChecksum
                                   , testSerializeEchoRequest
                                   , testSerializeEchoResponse ]

testIpChecksum :: TestTree
testIpChecksum = testCase "ipChecksum case 1" $ do
  let testIcmp = B.pack [
        0x08, 0x00, 0x00, 0x00, 0xfc, 0x6b, 0x59, 0xfb,
        0x61, 0x62, 0x63, 0x64, 0x65, 0x66, 0x67, 0x68]
  let checksum = ipChecksum testIcmp
  checksum @?= 0x1003

testSerializeEchoRequest :: TestTree
testSerializeEchoRequest = testCase "serialize IcmpEchoRequest" $ do
  let payload = B.pack [
        0x1c, 0x25, 0x31, 0x65, 0x00, 0x00, 0x00, 0x00,
        0x72, 0x51, 0x03, 0x00, 0x00, 0x00, 0x00, 0x00, 0x10, 0x11, 0x12, 0x13, 0x14, 0x15, 0x16, 0x17,
        0x18, 0x19, 0x1a, 0x1b, 0x1c, 0x1d, 0x1e, 0x1f, 0x20, 0x21, 0x22, 0x23, 0x24, 0x25, 0x26, 0x27,
        0x28, 0x29, 0x2a, 0x2b, 0x2c, 0x2d, 0x2e, 0x2f, 0x30, 0x31, 0x32, 0x33, 0x34, 0x35, 0x36, 0x37
        ]
  let testIcmp = B.pack [ 0x08, 0x00, 0xd8, 0x7c, 0x9d, 0xd2, 0x00, 0x02] <> payload
  let icmp = IcmpEchoRequest (IcmpId 40402) (IcmpSqnum 2) payload
  (B.toStrict . encode) icmp @=? testIcmp

testSerializeEchoResponse :: TestTree
testSerializeEchoResponse = testCase "serialize IcmpEchoResponse" $ do
  let payload = B.pack [
        0x1c, 0x25, 0x31, 0x65, 0x00, 0x00, 0x00, 0x00,
        0x72, 0x51, 0x03, 0x00, 0x00, 0x00, 0x00, 0x00, 0x10, 0x11, 0x12, 0x13, 0x14, 0x15, 0x16, 0x17,
        0x18, 0x19, 0x1a, 0x1b, 0x1c, 0x1d, 0x1e, 0x1f, 0x20, 0x21, 0x22, 0x23, 0x24, 0x25, 0x26, 0x27,
        0x28, 0x29, 0x2a, 0x2b, 0x2c, 0x2d, 0x2e, 0x2f, 0x30, 0x31, 0x32, 0x33, 0x34, 0x35, 0x36, 0x37
        ]
  let testIcmp = B.pack [ 0x00, 0x00, 0xe0, 0x7c, 0x9d, 0xd2, 0x00, 0x02] <> payload
  let icmp = IcmpEchoResponse (IcmpId 40402) (IcmpSqnum 2) payload
  (B.toStrict . encode) icmp @=? testIcmp