author | smxx |
Fri, 28 May 2010 12:47:17 +0000 | |
changeset 3481 | c385a7dad025 |
parent 3458 | 11cd56019f00 |
child 3500 | af8390d807d6 |
permissions | -rw-r--r-- |
2348 | 1 |
{-# LANGUAGE ScopedTypeVariables #-} |
1804 | 2 |
module ClientIO where |
3 |
||
2296
19f2f76dc346
Patch for compiling with 6.10 (define NEW_EXCEPTIONS to do that)
unc0rr
parents:
2126
diff
changeset
|
4 |
import qualified Control.Exception as Exception |
1804 | 5 |
import Control.Concurrent.Chan |
2867
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2352
diff
changeset
|
6 |
import Control.Concurrent |
1804 | 7 |
import Control.Monad |
8 |
import System.IO |
|
2952
18fada739b55
- Convert strings from utf-8 on recieve, and back to utf-8 when send them
unc0rr
parents:
2867
diff
changeset
|
9 |
import qualified Data.ByteString.UTF8 as BUTF8 |
18fada739b55
- Convert strings from utf-8 on recieve, and back to utf-8 when send them
unc0rr
parents:
2867
diff
changeset
|
10 |
import qualified Data.ByteString as B |
1804 | 11 |
---------------- |
12 |
import CoreTypes |
|
3435 | 13 |
import RoomsAndClients |
1804 | 14 |
|
3435 | 15 |
listenLoop :: Handle -> Int -> [String] -> Chan CoreMessage -> ClientIndex -> IO () |
2001 | 16 |
listenLoop handle linesNumber buf chan clientID = do |
3458 | 17 |
putStrLn $ show handle ++ show buf ++ show clientID |
2952
18fada739b55
- Convert strings from utf-8 on recieve, and back to utf-8 when send them
unc0rr
parents:
2867
diff
changeset
|
18 |
str <- liftM BUTF8.toString $ B.hGetLine handle |
2867
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2352
diff
changeset
|
19 |
if (linesNumber > 50) || (length str > 450) then |
3458 | 20 |
protocolViolationMsg >> freeClient |
2867
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2352
diff
changeset
|
21 |
else |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2352
diff
changeset
|
22 |
if str == "" then do |
3458 | 23 |
writeChan chan $ ClientMessage (clientID, reverse buf) |
2867
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2352
diff
changeset
|
24 |
yield |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2352
diff
changeset
|
25 |
listenLoop handle 0 [] chan clientID |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2352
diff
changeset
|
26 |
else |
3458 | 27 |
listenLoop handle (linesNumber + 1) (str : buf) chan clientID |
28 |
where |
|
29 |
protocolViolationMsg = writeChan chan $ ClientMessage (clientID, ["QUIT", "Protocol violation"]) |
|
30 |
freeClient = writeChan chan $ FreeClient clientID |
|
31 |
||
1804 | 32 |
|
3435 | 33 |
clientRecvLoop :: Handle -> Chan CoreMessage -> ClientIndex -> IO () |
1804 | 34 |
clientRecvLoop handle chan clientID = |
2867
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2352
diff
changeset
|
35 |
listenLoop handle 0 [] chan clientID |
3458 | 36 |
`catch` (\e -> clientOff (show e) >> freeClient >> return ()) |
37 |
where |
|
38 |
clientOff msg = writeChan chan $ ClientMessage (clientID, ["QUIT", msg]) -- if the client disconnects, we perform as if it sent QUIT message |
|
39 |
freeClient = writeChan chan $ FreeClient clientID |
|
1804 | 40 |
|
3435 | 41 |
clientSendLoop :: Handle -> Chan CoreMessage -> Chan [String] -> ClientIndex -> IO() |
1804 | 42 |
clientSendLoop handle coreChan chan clientID = do |
2867
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2352
diff
changeset
|
43 |
answer <- readChan chan |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2352
diff
changeset
|
44 |
doClose <- Exception.handle |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2352
diff
changeset
|
45 |
(\(e :: Exception.IOException) -> if isQuit answer then return True else sendQuit e >> return False) $ do |
2954 | 46 |
B.hPutStrLn handle $ BUTF8.fromString $ unlines answer |
47 |
hFlush handle |
|
48 |
return $ isQuit answer |
|
1804 | 49 |
|
2867
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2352
diff
changeset
|
50 |
if doClose then |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2352
diff
changeset
|
51 |
Exception.handle (\(_ :: Exception.IOException) -> putStrLn "error on hClose") $ hClose handle |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2352
diff
changeset
|
52 |
else |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2352
diff
changeset
|
53 |
clientSendLoop handle coreChan chan clientID |
1804 | 54 |
|
2867
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2352
diff
changeset
|
55 |
where |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2352
diff
changeset
|
56 |
sendQuit e = writeChan coreChan $ ClientMessage (clientID, ["QUIT", show e]) |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2352
diff
changeset
|
57 |
isQuit ("BYE":xs) = True |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2352
diff
changeset
|
58 |
isQuit _ = False |