ArtificialFields.hs

{-|
Module      : MptcpAnalyzer.ArtificialFields
Description : A set of artifical fields to ease dataframe processing
Maintainer  : matt


generated in a postprocess step

-}
{-# LANGUAGE DeriveAnyClass #-}
{-# LANGUAGE DeriveGeneric #-}
{-# LANGUAGE DerivingStrategies #-}
{-# LANGUAGE OverloadedStrings #-}
module MptcpAnalyzer.ArtificialFields
where
import MptcpAnalyzer.Stream

import Net.IP
-- import Net.IPv6 (IPv6(..))
import GHC.TypeLits (KnownSymbol)
-- import Language.Haskell.TH (Name)
import Data.Map (Map, fromList)
import Data.String
import Data.Text (Text)
import Data.Word (Word16, Word32, Word64, Word8)
import Frames.ShowCSV
import Language.Haskell.TH (Name)
import Options.Applicative
import Tshark.Fields

-- | Filters a connection depending on its role
data ConnectionRole = RoleServer | RoleClient deriving (Show, Eq, Enum, Read, ShowCSV, Ord)

showConnectionRole :: (IsString a) => ConnectionRole -> a
showConnectionRole RoleServer = "Server"
showConnectionRole RoleClient = "Client"

artificialFields :: FieldDescriptions
artificialFields = fromList [
    ("tcpDest", TsharkFieldDesc "" ''ConnectionRole Nothing False)
    , ("mptcpDest", TsharkFieldDesc "" ''ConnectionRole Nothing False)
    , ("conDest", TsharkFieldDesc "" ''ConnectionRole Nothing False)
    , ("packetHash", TsharkFieldDesc "" ''ConnectionRole Nothing False)
  ]

-- TODO remove
-- mergedFields :: [(Text, Name)]
-- mergedFields = [
--   ("senderIP", ''IP)
--   , ("receiverIP", ''IP)
--   , ("sndTime", ''Double)
--   , ("rcvTime", ''Double)
--   , ("tcpSeq", ''Word32)
--   ]


readConnectionRole :: ReadM ConnectionRole
readConnectionRole = eitherReader $ \arg -> case arg of
  "server" -> return RoleServer
  "client" -> return RoleClient
  -- [("client", "")] -> return $ RoleClient
  _other -> Left $ "readConnectionRole: cannot parse value `" ++ arg ++ "`. It must be either 'server' or 'client'"