author | unc0rr |
Fri, 27 Mar 2009 18:50:18 +0000 | |
changeset 1926 | cb46fbdcaa41 |
parent 1841 | fba7210b438b |
child 1927 | e2031906a347 |
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 |
|
5dd4cb7fd7e5
Server now send ASKPASSWORD command to frontend when user has web account
unc0rr
parents:
1833
diff
changeset
|
19 |
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
|
20 |
|
1804 | 21 |
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
|
22 |
reactCmd serverInfo clID cmd clients rooms = |
fba7210b438b
Retrieve client password from web database and ask for it
unc0rr
parents:
1839
diff
changeset
|
23 |
liftM firstAway $ foldM processAction (clID, serverInfo, clients, rooms) $ handleCmd clID clients rooms cmd |
1804 | 24 |
|
1839
5dd4cb7fd7e5
Server now send ASKPASSWORD command to frontend when user has web account
unc0rr
parents:
1833
diff
changeset
|
25 |
mainLoop :: ServerInfo -> Clients -> Rooms -> IO () |
5dd4cb7fd7e5
Server now send ASKPASSWORD command to frontend when user has web account
unc0rr
parents:
1833
diff
changeset
|
26 |
mainLoop serverInfo clients rooms = do |
5dd4cb7fd7e5
Server now send ASKPASSWORD command to frontend when user has web account
unc0rr
parents:
1833
diff
changeset
|
27 |
r <- readChan $ coreChan serverInfo |
1804 | 28 |
|
29 |
(newServerInfo, mClients, mRooms) <- |
|
30 |
case r of |
|
31 |
Accept ci -> do |
|
1841
fba7210b438b
Retrieve client password from web database and ask for it
unc0rr
parents:
1839
diff
changeset
|
32 |
liftM firstAway $ processAction |
1926
cb46fbdcaa41
Add simple DoS protection mechanism (although better than previous server had)
unc0rr
parents:
1841
diff
changeset
|
33 |
(clientUID ci, serverInfo, clients, rooms) (AddClient ci) |
1804 | 34 |
|
35 |
ClientMessage (clID, cmd) -> do |
|
36 |
debugM "Clients" $ (show clID) ++ ": " ++ (show cmd) |
|
37 |
if clID `IntMap.member` clients then |
|
38 |
reactCmd serverInfo clID cmd clients rooms |
|
39 |
else |
|
40 |
do |
|
41 |
debugM "Clients" "Message from dead client" |
|
42 |
return (serverInfo, clients, rooms) |
|
43 |
||
1839
5dd4cb7fd7e5
Server now send ASKPASSWORD command to frontend when user has web account
unc0rr
parents:
1833
diff
changeset
|
44 |
ClientAccountInfo clID info -> |
5dd4cb7fd7e5
Server now send ASKPASSWORD command to frontend when user has web account
unc0rr
parents:
1833
diff
changeset
|
45 |
if clID `IntMap.member` clients then |
5dd4cb7fd7e5
Server now send ASKPASSWORD command to frontend when user has web account
unc0rr
parents:
1833
diff
changeset
|
46 |
liftM firstAway $ processAction |
5dd4cb7fd7e5
Server now send ASKPASSWORD command to frontend when user has web account
unc0rr
parents:
1833
diff
changeset
|
47 |
(clID, serverInfo, clients, rooms) |
5dd4cb7fd7e5
Server now send ASKPASSWORD command to frontend when user has web account
unc0rr
parents:
1833
diff
changeset
|
48 |
(ProcessAccountInfo info) |
5dd4cb7fd7e5
Server now send ASKPASSWORD command to frontend when user has web account
unc0rr
parents:
1833
diff
changeset
|
49 |
else |
5dd4cb7fd7e5
Server now send ASKPASSWORD command to frontend when user has web account
unc0rr
parents:
1833
diff
changeset
|
50 |
do |
5dd4cb7fd7e5
Server now send ASKPASSWORD command to frontend when user has web account
unc0rr
parents:
1833
diff
changeset
|
51 |
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
|
52 |
return (serverInfo, clients, rooms) |
5dd4cb7fd7e5
Server now send ASKPASSWORD command to frontend when user has web account
unc0rr
parents:
1833
diff
changeset
|
53 |
|
5dd4cb7fd7e5
Server now send ASKPASSWORD command to frontend when user has web account
unc0rr
parents:
1833
diff
changeset
|
54 |
|
1804 | 55 |
{- let hadRooms = (not $ null rooms) && (null mrooms) |
56 |
in unless ((not $ isDedicated serverInfo) && ((null clientsIn) || hadRooms)) $ |
|
57 |
mainLoop serverInfo acceptChan messagesChan clientsIn mrooms -} |
|
58 |
||
1839
5dd4cb7fd7e5
Server now send ASKPASSWORD command to frontend when user has web account
unc0rr
parents:
1833
diff
changeset
|
59 |
mainLoop newServerInfo mClients mRooms |
1804 | 60 |
|
61 |
startServer :: ServerInfo -> Chan CoreMessage -> Socket -> IO () |
|
62 |
startServer serverInfo coreChan serverSocket = do |
|
63 |
putStrLn $ "Listening on port " ++ show (listenPort serverInfo) |
|
64 |
||
65 |
forkIO $ |
|
66 |
acceptLoop |
|
67 |
serverSocket |
|
68 |
coreChan |
|
69 |
0 |
|
70 |
||
71 |
return () |
|
72 |
||
73 |
{- forkIO $ messagesLoop messagesChan |
|
74 |
forkIO $ timerLoop messagesChan-} |
|
75 |
||
1833 | 76 |
startDBConnection $ serverInfo |
1804 | 77 |
|
1839
5dd4cb7fd7e5
Server now send ASKPASSWORD command to frontend when user has web account
unc0rr
parents:
1833
diff
changeset
|
78 |
mainLoop serverInfo IntMap.empty (IntMap.singleton 0 newRoom) |