mirror of
https://github.com/simplex-chat/simplexmq.git
synced 2026-06-07 13:12:10 +00:00
Evgeny
f3408d9bb6
* explicit exports * more empty exports * add exports * reorder * use correct ControlProtocol type for xftp router --------- Co-authored-by: Evgeny @ SimpleX Chat <259188159+evgeny-simplex@users.noreply.github.com>
74 lines
1.8 KiB
Haskell
74 lines
1.8 KiB
Haskell
{-# LANGUAGE CApiFFI #-}
|
|
{-# LANGUAGE CPP #-}
|
|
{-# LANGUAGE DeriveAnyClass #-}
|
|
{-# LANGUAGE NamedFieldPuns #-}
|
|
{-# LANGUAGE TemplateHaskell #-}
|
|
|
|
module Simplex.Messaging.Transport.KeepAlive
|
|
( KeepAliveOpts (..),
|
|
defaultKeepAliveOpts,
|
|
setSocketKeepAlive,
|
|
) where
|
|
|
|
import qualified Data.Aeson.TH as J
|
|
import Foreign.C (CInt (..))
|
|
import Network.Socket
|
|
import Simplex.Messaging.Parsers (defaultJSON)
|
|
|
|
data KeepAliveOpts = KeepAliveOpts
|
|
{ keepIdle :: Int,
|
|
keepIntvl :: Int,
|
|
keepCnt :: Int
|
|
}
|
|
deriving (Eq, Show)
|
|
|
|
defaultKeepAliveOpts :: KeepAliveOpts
|
|
defaultKeepAliveOpts =
|
|
KeepAliveOpts
|
|
{ keepIdle = 30,
|
|
keepIntvl = 15,
|
|
keepCnt = 4
|
|
}
|
|
|
|
_SOL_TCP :: CInt
|
|
_SOL_TCP = 6
|
|
|
|
#if defined(mingw32_HOST_OS)
|
|
-- Windows
|
|
|
|
-- The values are copied from windows::Win32::Networking::WinSock
|
|
-- https://microsoft.github.io/windows-docs-rs/doc/windows/Win32/Networking/WinSock/index.html
|
|
|
|
_TCP_KEEPIDLE :: CInt
|
|
_TCP_KEEPIDLE = 3
|
|
|
|
_TCP_KEEPINTVL :: CInt
|
|
_TCP_KEEPINTVL = 17
|
|
|
|
_TCP_KEEPCNT :: CInt
|
|
_TCP_KEEPCNT = 16
|
|
|
|
#else
|
|
-- Mac/Linux
|
|
|
|
#if defined(darwin_HOST_OS)
|
|
foreign import capi "netinet/tcp.h value TCP_KEEPALIVE" _TCP_KEEPIDLE :: CInt
|
|
#else
|
|
foreign import capi "netinet/tcp.h value TCP_KEEPIDLE" _TCP_KEEPIDLE :: CInt
|
|
#endif
|
|
|
|
foreign import capi "netinet/tcp.h value TCP_KEEPINTVL" _TCP_KEEPINTVL :: CInt
|
|
|
|
foreign import capi "netinet/tcp.h value TCP_KEEPCNT" _TCP_KEEPCNT :: CInt
|
|
|
|
#endif
|
|
|
|
setSocketKeepAlive :: Socket -> KeepAliveOpts -> IO ()
|
|
setSocketKeepAlive sock KeepAliveOpts {keepCnt, keepIdle, keepIntvl} = do
|
|
setSocketOption sock KeepAlive 1
|
|
setSocketOption sock (SockOpt _SOL_TCP _TCP_KEEPIDLE) keepIdle
|
|
setSocketOption sock (SockOpt _SOL_TCP _TCP_KEEPINTVL) keepIntvl
|
|
setSocketOption sock (SockOpt _SOL_TCP _TCP_KEEPCNT) keepCnt
|
|
|
|
$(J.deriveJSON defaultJSON ''KeepAliveOpts)
|