diff -r e848447f29be -r ce26e16d18ab netserver/newhwserv.hs --- a/netserver/newhwserv.hs Sun Oct 05 23:36:11 2008 +0000 +++ b/netserver/newhwserv.hs Mon Oct 06 18:07:38 2008 +0000 @@ -5,7 +5,7 @@ import System.IO import Control.Concurrent import Control.Concurrent.STM -import Control.Exception (finally) +import Control.Exception (setUncaughtExceptionHandler, handle, finally) import Control.Monad (forM, forM_, filterM, liftM) import Data.List import Miscutils @@ -20,6 +20,7 @@ hPutStrLn cHandle "CONNECTED\n" acceptLoop servSock acceptChan + listenLoop :: Handle -> [String] -> TChan [String] -> IO () listenLoop handle buf chan = do str <- hGetLine handle @@ -29,31 +30,32 @@ else listenLoop handle (buf ++ [str]) chan + clientLoop :: Handle -> TChan [String] -> IO () clientLoop handle chan = listenLoop handle [] chan `catch` (const $ clientOff >> return ()) where clientOff = atomically $ writeTChan chan ["QUIT"] -- если клиент отключается, то делаем вид, что от него пришла команда QUIT + sendAnswers [] _ clients _ = return clients sendAnswers ((handlesFunc, answer):answers) client clients rooms = do - putStrLn ("< " ++ show answer) - let recipients = handlesFunc client clients rooms + putStrLn ("< " ++ (show answer) ++ " (" ++ (show $ length recipients) ++ " recipients)") clHandles' <- forM recipients $ - \ch -> do + \ch -> Control.Exception.handle (\e -> putStrLn (show e) >> hClose ch >> return [ch]) $ + do forM_ answer (\str -> hPutStrLn ch str) hPutStrLn ch "" hFlush ch if (not $ null answer) && (head answer == "BYE") then hClose ch >> return [ch] else return [] - `catch` const (hClose ch >> return [ch]) let mclients = remove clients $ concat clHandles' sendAnswers answers client mclients rooms where - remove list rmClHandles = deleteFirstsBy2t (\ a b -> (handle a) == b) list rmClHandles + remove list rmClHandles = deleteFirstsBy2t (\ a b -> (Miscutils.handle a) == b) list rmClHandles mainLoop :: Socket -> TChan ClientInfo -> [ClientInfo] -> [RoomInfo] -> IO () @@ -64,20 +66,21 @@ mainLoop servSock acceptChan (ci:clients) rooms Right (cmd, client) -> do putStrLn ("> " ++ show cmd) + let (clientsFunc, roomsFunc, answers) = handleCmd client clients rooms $ cmd - - let mclients = clientsFunc clients let mrooms = roomsFunc rooms - mclients <- sendAnswers answers client clients rooms + clientsIn <- sendAnswers answers client (clientsFunc clients) mrooms - mainLoop servSock acceptChan mclients mrooms + mainLoop servSock acceptChan clientsIn mrooms + startServer serverSocket = do acceptChan <- atomically newTChan forkIO $ acceptLoop serverSocket acceptChan mainLoop serverSocket acceptChan [] [] + main = withSocketsDo $ do serverSocket <- listenOn $ Service "hedgewars" startServer serverSocket `finally` sClose serverSocket