forked from schmittlauch/Hash2Pub
39 lines
1.4 KiB
Haskell
39 lines
1.4 KiB
Haskell
{-# LANGUAGE OverloadedStrings #-}
|
|
|
|
module Main where
|
|
|
|
import Network.Socket hiding (send, sendTo, recv, recvFrom)
|
|
import Network.Socket.ByteString
|
|
import Control.Concurrent
|
|
import Control.Exception
|
|
import Control.Monad (forever)
|
|
import System.IO (
|
|
IOMode (ReadWriteMode)
|
|
)
|
|
import qualified Data.ByteString as BS
|
|
|
|
main = do
|
|
sock <- socket AF_INET6 Datagram defaultProtocol
|
|
setSocketOption sock IPv6Only 1
|
|
-- lookup destination to get a SockAddr
|
|
let hints = defaultHints { addrFamily = AF_INET6, addrSocketType = Datagram}
|
|
serverAddr <- addrAddress . head <$> getAddrInfo (Just hints) (Just "::1") (Just "7331")
|
|
print serverAddr
|
|
bind sock serverAddr
|
|
print sock
|
|
forever $ do
|
|
receivedStuff <- recvFrom sock 65535 -- blocks
|
|
forkIO $ bracket newSendSocket close (serveReceive receivedStuff)
|
|
|
|
serveReceive :: (BS.ByteString, SockAddr) -> Socket -> IO ()
|
|
serveReceive (msg, fromAddr) sendSocket = do
|
|
print sendSocket
|
|
putStrLn $ "Got message " ++ show msg ++ " from " ++ show fromAddr
|
|
-- important: as new socket sends from a different port, the receiving client must use an unconnected socket!
|
|
sentBytes <- sendTo sendSocket ("Hi, thx for " `BS.append` msg) fromAddr
|
|
putStrLn $ "sent response of " ++ show sentBytes ++ "bytes"
|
|
return ()
|
|
|
|
newSendSocket :: IO Socket
|
|
newSendSocket = socket AF_INET6 Datagram defaultProtocol
|