author | unc0rr |
Fri, 12 Nov 2010 19:48:51 +0300 | |
branch | mbait-mangen |
changeset 2566 | 4fb45d86d5e8 |
parent 2349 | ba7a0813c532 |
child 2867 | 9be6693c78cb |
permissions | -rw-r--r-- |
1804 | 1 |
module ServerCore where |
2 |
||
3 |
import Network |
|
4 |
import Control.Concurrent |
|
5 |
import Control.Concurrent.STM |
|
6 |
import Control.Concurrent.Chan |
|
7 |
import Control.Monad |
|
8 |
import qualified Data.IntMap as IntMap |
|
9 |
import System.Log.Logger |
|
10 |
-------------------------------------- |
|
11 |
import CoreTypes |
|
12 |
import NetRoutines |
|
13 |
import Utils |
|
14 |
import HWProtoCore |
|
15 |
import Actions |
|
1833 | 16 |
import OfficialServer.DBInteraction |
1804 | 17 |
|
1839
5dd4cb7fd7e5
Server now send ASKPASSWORD command to frontend when user has web account
unc0rr
parents:
1833
diff
changeset
|
18 |
|
2173 | 19 |
timerLoop :: Int -> Chan CoreMessage -> IO() |
2349 | 20 |
timerLoop tick messagesChan = threadDelay (30 * 10^6) >> writeChan messagesChan (TimerAction tick) >> timerLoop (tick + 1) messagesChan |
1927
e2031906a347
Ping clients every 30 seconds. Disconnection due to ping timeout to be implemented.
unc0rr
parents:
1926
diff
changeset
|
21 |
|
1839
5dd4cb7fd7e5
Server now send ASKPASSWORD command to frontend when user has web account
unc0rr
parents:
1833
diff
changeset
|
22 |
firstAway (_, a, b, c) = (a, b, c) |
5dd4cb7fd7e5
Server now send ASKPASSWORD command to frontend when user has web account
unc0rr
parents:
1833
diff
changeset
|
23 |
|
1804 | 24 |
reactCmd :: ServerInfo -> Int -> [String] -> Clients -> Rooms -> IO (ServerInfo, Clients, Rooms) |
1841
fba7210b438b
Retrieve client password from web database and ask for it
unc0rr
parents:
1839
diff
changeset
|
25 |
reactCmd serverInfo clID cmd clients rooms = |
fba7210b438b
Retrieve client password from web database and ask for it
unc0rr
parents:
1839
diff
changeset
|
26 |
liftM firstAway $ foldM processAction (clID, serverInfo, clients, rooms) $ handleCmd clID clients rooms cmd |
1804 | 27 |
|
1839
5dd4cb7fd7e5
Server now send ASKPASSWORD command to frontend when user has web account
unc0rr
parents:
1833
diff
changeset
|
28 |
mainLoop :: ServerInfo -> Clients -> Rooms -> IO () |
5dd4cb7fd7e5
Server now send ASKPASSWORD command to frontend when user has web account
unc0rr
parents:
1833
diff
changeset
|
29 |
mainLoop serverInfo clients rooms = do |
5dd4cb7fd7e5
Server now send ASKPASSWORD command to frontend when user has web account
unc0rr
parents:
1833
diff
changeset
|
30 |
r <- readChan $ coreChan serverInfo |
1804 | 31 |
|
32 |
(newServerInfo, mClients, mRooms) <- |
|
33 |
case r of |
|
2349 | 34 |
Accept ci -> |
1841
fba7210b438b
Retrieve client password from web database and ask for it
unc0rr
parents:
1839
diff
changeset
|
35 |
liftM firstAway $ processAction |
1926
cb46fbdcaa41
Add simple DoS protection mechanism (although better than previous server had)
unc0rr
parents:
1841
diff
changeset
|
36 |
(clientUID ci, serverInfo, clients, rooms) (AddClient ci) |
1804 | 37 |
|
38 |
ClientMessage (clID, cmd) -> do |
|
39 |
debugM "Clients" $ (show clID) ++ ": " ++ (show cmd) |
|
40 |
if clID `IntMap.member` clients then |
|
41 |
reactCmd serverInfo clID cmd clients rooms |
|
42 |
else |
|
43 |
do |
|
44 |
debugM "Clients" "Message from dead client" |
|
45 |
return (serverInfo, clients, rooms) |
|
46 |
||
2116
dec7ead2d178
Bring back authentication to official server, now using separate process to perform database interaction
unc0rr
parents:
1927
diff
changeset
|
47 |
ClientAccountInfo (clID, info) -> |
1839
5dd4cb7fd7e5
Server now send ASKPASSWORD command to frontend when user has web account
unc0rr
parents:
1833
diff
changeset
|
48 |
if clID `IntMap.member` clients then |
5dd4cb7fd7e5
Server now send ASKPASSWORD command to frontend when user has web account
unc0rr
parents:
1833
diff
changeset
|
49 |
liftM firstAway $ processAction |
5dd4cb7fd7e5
Server now send ASKPASSWORD command to frontend when user has web account
unc0rr
parents:
1833
diff
changeset
|
50 |
(clID, serverInfo, clients, rooms) |
5dd4cb7fd7e5
Server now send ASKPASSWORD command to frontend when user has web account
unc0rr
parents:
1833
diff
changeset
|
51 |
(ProcessAccountInfo info) |
5dd4cb7fd7e5
Server now send ASKPASSWORD command to frontend when user has web account
unc0rr
parents:
1833
diff
changeset
|
52 |
else |
5dd4cb7fd7e5
Server now send ASKPASSWORD command to frontend when user has web account
unc0rr
parents:
1833
diff
changeset
|
53 |
do |
5dd4cb7fd7e5
Server now send ASKPASSWORD command to frontend when user has web account
unc0rr
parents:
1833
diff
changeset
|
54 |
debugM "Clients" "Got info for dead client" |
5dd4cb7fd7e5
Server now send ASKPASSWORD command to frontend when user has web account
unc0rr
parents:
1833
diff
changeset
|
55 |
return (serverInfo, clients, rooms) |
1927
e2031906a347
Ping clients every 30 seconds. Disconnection due to ping timeout to be implemented.
unc0rr
parents:
1926
diff
changeset
|
56 |
|
2173 | 57 |
TimerAction tick -> |
2172 | 58 |
liftM firstAway $ |
2173 | 59 |
foldM processAction (0, serverInfo, clients, rooms) $ |
2349 | 60 |
PingAll : [StatsAction | even tick] |
2172 | 61 |
|
1839
5dd4cb7fd7e5
Server now send ASKPASSWORD command to frontend when user has web account
unc0rr
parents:
1833
diff
changeset
|
62 |
|
1804 | 63 |
{- let hadRooms = (not $ null rooms) && (null mrooms) |
64 |
in unless ((not $ isDedicated serverInfo) && ((null clientsIn) || hadRooms)) $ |
|
65 |
mainLoop serverInfo acceptChan messagesChan clientsIn mrooms -} |
|
66 |
||
1839
5dd4cb7fd7e5
Server now send ASKPASSWORD command to frontend when user has web account
unc0rr
parents:
1833
diff
changeset
|
67 |
mainLoop newServerInfo mClients mRooms |
1804 | 68 |
|
1927
e2031906a347
Ping clients every 30 seconds. Disconnection due to ping timeout to be implemented.
unc0rr
parents:
1926
diff
changeset
|
69 |
startServer :: ServerInfo -> Socket -> IO () |
e2031906a347
Ping clients every 30 seconds. Disconnection due to ping timeout to be implemented.
unc0rr
parents:
1926
diff
changeset
|
70 |
startServer serverInfo serverSocket = do |
1804 | 71 |
putStrLn $ "Listening on port " ++ show (listenPort serverInfo) |
72 |
||
73 |
forkIO $ |
|
74 |
acceptLoop |
|
75 |
serverSocket |
|
1927
e2031906a347
Ping clients every 30 seconds. Disconnection due to ping timeout to be implemented.
unc0rr
parents:
1926
diff
changeset
|
76 |
(coreChan serverInfo) |
1804 | 77 |
0 |
78 |
||
79 |
return () |
|
80 |
||
2173 | 81 |
forkIO $ timerLoop 0 $ coreChan serverInfo |
1804 | 82 |
|
2349 | 83 |
startDBConnection serverInfo |
1804 | 84 |
|
2184 | 85 |
forkIO $ mainLoop serverInfo IntMap.empty (IntMap.singleton 0 newRoom) |
86 |
||
87 |
forever $ threadDelay (60 * 60 * 10^6) >> putStrLn "***" |