author | koda |
Wed, 22 Sep 2010 02:22:15 +0200 | |
changeset 3893 | 568bfd083465 |
parent 3741 | 73246d25dfe1 |
child 3947 | 709fdb89f76c |
permissions | -rw-r--r-- |
3500
af8390d807d6
Use sockets instead of handles, use bytestrings instead of strings
unc0rr
parents:
3458
diff
changeset
|
1 |
{-# LANGUAGE OverloadedStrings #-} |
1804 | 2 |
module Actions where |
3 |
||
3425 | 4 |
import Control.Concurrent |
1804 | 5 |
import Control.Concurrent.Chan |
6 |
import qualified Data.IntSet as IntSet |
|
3566 | 7 |
import qualified Data.Set as Set |
1813 | 8 |
import qualified Data.Sequence as Seq |
1839
5dd4cb7fd7e5
Server now send ASKPASSWORD command to frontend when user has web account
unc0rr
parents:
1834
diff
changeset
|
9 |
import System.Log.Logger |
3671
a94d1dc4a8d9
- burp's patch cleaning up module dependancies + cabal file
unc0rr
parents:
3658
diff
changeset
|
10 |
import Control.Monad |
1926
cb46fbdcaa41
Add simple DoS protection mechanism (although better than previous server had)
unc0rr
parents:
1925
diff
changeset
|
11 |
import Data.Time |
3671
a94d1dc4a8d9
- burp's patch cleaning up module dependancies + cabal file
unc0rr
parents:
3658
diff
changeset
|
12 |
import Data.Maybe |
3435 | 13 |
import Control.Monad.Reader |
3741
73246d25dfe1
Add some more strictness, use unsafeThaw and unsafeFreeze functions which work at O(1)
unc0rr
parents:
3673
diff
changeset
|
14 |
import Control.Monad.State.Strict |
3502 | 15 |
import qualified Data.ByteString.Char8 as B |
1804 | 16 |
----------------------------- |
17 |
import CoreTypes |
|
1813 | 18 |
import Utils |
3425 | 19 |
import ClientIO |
3458 | 20 |
import ServerState |
1804 | 21 |
|
22 |
data Action = |
|
3566 | 23 |
AnswerClients ![ClientChan] ![B.ByteString] |
2867
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
24 |
| SendServerMessage |
3283 | 25 |
| SendServerVars |
3502 | 26 |
| MoveToRoom RoomIndex |
3531 | 27 |
| MoveToLobby B.ByteString |
3502 | 28 |
| RemoveTeam B.ByteString |
2867
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
29 |
| RemoveRoom |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
30 |
| UnreadyRoomClients |
3502 | 31 |
| JoinLobby |
32 |
| ProtocolError B.ByteString |
|
33 |
| Warning B.ByteString |
|
34 |
| ByeClient B.ByteString |
|
35 |
| KickClient ClientIndex |
|
36 |
| KickRoomClient ClientIndex |
|
37 |
| BanClient B.ByteString -- nick |
|
38 |
| RemoveClientTeams ClientIndex |
|
2867
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
39 |
| ModifyClient (ClientInfo -> ClientInfo) |
3458 | 40 |
| ModifyClient2 ClientIndex (ClientInfo -> ClientInfo) |
2867
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
41 |
| ModifyRoom (RoomInfo -> RoomInfo) |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
42 |
| ModifyServerInfo (ServerInfo -> ServerInfo) |
3502 | 43 |
| AddRoom B.ByteString B.ByteString |
2867
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
44 |
| CheckRegistered |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
45 |
| ClearAccountsCache |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
46 |
| ProcessAccountInfo AccountInfo |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
47 |
| Dump |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
48 |
| AddClient ClientInfo |
3566 | 49 |
| DeleteClient ClientIndex |
2867
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
50 |
| PingAll |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
51 |
| StatsAction |
1804 | 52 |
|
3502 | 53 |
type CmdHandler = [B.ByteString] -> Reader (ClientIndex, IRnC) [Action] |
1804 | 54 |
|
3451
62089ccec75c
Uses StateT monad instead of manually maintaining the state
unc0rr
parents:
3436
diff
changeset
|
55 |
|
3458 | 56 |
processAction :: Action -> StateT ServerState IO () |
1804 | 57 |
|
58 |
||
3673
45778b16b224
Some comments on the reason of the bug, leave bug not fixed yet
unc0rr
parents:
3671
diff
changeset
|
59 |
processAction (AnswerClients chans msg) = do |
3741
73246d25dfe1
Add some more strictness, use unsafeThaw and unsafeFreeze functions which work at O(1)
unc0rr
parents:
3673
diff
changeset
|
60 |
liftIO $ map (flip seq ()) chans `seq` map (flip seq ()) msg `seq` mapM_ (flip writeChan msg) chans |
1804 | 61 |
|
62 |
||
3501 | 63 |
processAction SendServerMessage = do |
64 |
chan <- client's sendChan |
|
65 |
protonum <- client's clientProto |
|
66 |
si <- liftM serverInfo get |
|
67 |
let message = if protonum < latestReleaseVersion si then |
|
3283 | 68 |
serverMessageForOldVersions si |
2867
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
69 |
else |
3283 | 70 |
serverMessage si |
3673
45778b16b224
Some comments on the reason of the bug, leave bug not fixed yet
unc0rr
parents:
3671
diff
changeset
|
71 |
processAction $ AnswerClients [chan] ["SERVER_MESSAGE", message] |
3501 | 72 |
{- |
3283 | 73 |
|
3425 | 74 |
processAction (clID, serverInfo, rnc) SendServerVars = do |
3283 | 75 |
writeChan (sendChan $ clients ! clID) ("SERVER_VARS" : vars) |
3425 | 76 |
return (clID, serverInfo, rnc) |
3283 | 77 |
where |
78 |
client = clients ! clID |
|
79 |
vars = [ |
|
3425 | 80 |
"MOTD_NEW", serverMessage serverInfo, |
81 |
"MOTD_OLD", serverMessageForOldVersions serverInfo, |
|
3283 | 82 |
"LATEST_PROTO", show $ latestReleaseVersion serverInfo |
83 |
] |
|
1923 | 84 |
|
85 |
||
3501 | 86 |
-} |
87 |
||
88 |
processAction (ProtocolError msg) = do |
|
89 |
chan <- client's sendChan |
|
3673
45778b16b224
Some comments on the reason of the bug, leave bug not fixed yet
unc0rr
parents:
3671
diff
changeset
|
90 |
processAction $ AnswerClients [chan] ["ERROR", msg] |
1804 | 91 |
|
92 |
||
3501 | 93 |
processAction (Warning msg) = do |
94 |
chan <- client's sendChan |
|
3673
45778b16b224
Some comments on the reason of the bug, leave bug not fixed yet
unc0rr
parents:
3671
diff
changeset
|
95 |
processAction $ AnswerClients [chan] ["WARNING", msg] |
1804 | 96 |
|
3451
62089ccec75c
Uses StateT monad instead of manually maintaining the state
unc0rr
parents:
3436
diff
changeset
|
97 |
processAction (ByeClient msg) = do |
62089ccec75c
Uses StateT monad instead of manually maintaining the state
unc0rr
parents:
3436
diff
changeset
|
98 |
(Just ci) <- gets clientIndex |
62089ccec75c
Uses StateT monad instead of manually maintaining the state
unc0rr
parents:
3436
diff
changeset
|
99 |
rnc <- gets roomsClients |
62089ccec75c
Uses StateT monad instead of manually maintaining the state
unc0rr
parents:
3436
diff
changeset
|
100 |
ri <- clientRoomA |
3436
288fcbdb77b6
Make server build again (it's still useless though)
unc0rr
parents:
3435
diff
changeset
|
101 |
when (ri /= lobbyId) $ do |
3531 | 102 |
processAction $ MoveToLobby ("quit: " `B.append` msg) |
3436
288fcbdb77b6
Make server build again (it's still useless though)
unc0rr
parents:
3435
diff
changeset
|
103 |
return () |
1929 | 104 |
|
3501 | 105 |
chan <- client's sendChan |
3566 | 106 |
ready <- client's isReady |
3458 | 107 |
|
3451
62089ccec75c
Uses StateT monad instead of manually maintaining the state
unc0rr
parents:
3436
diff
changeset
|
108 |
liftIO $ do |
3500
af8390d807d6
Use sockets instead of handles, use bytestrings instead of strings
unc0rr
parents:
3458
diff
changeset
|
109 |
infoM "Clients" (show ci ++ " quits: " ++ (B.unpack msg)) |
3451
62089ccec75c
Uses StateT monad instead of manually maintaining the state
unc0rr
parents:
3436
diff
changeset
|
110 |
|
62089ccec75c
Uses StateT monad instead of manually maintaining the state
unc0rr
parents:
3436
diff
changeset
|
111 |
--mapM_ (processAction (ci, serverInfo, rnc)) $ answerOthersQuit ++ answerInformRoom |
62089ccec75c
Uses StateT monad instead of manually maintaining the state
unc0rr
parents:
3436
diff
changeset
|
112 |
modifyRoom rnc (\r -> r{ |
62089ccec75c
Uses StateT monad instead of manually maintaining the state
unc0rr
parents:
3436
diff
changeset
|
113 |
--playersIDs = IntSet.delete ci (playersIDs r) |
3566 | 114 |
playersIn = (playersIn r) - 1, |
115 |
readyPlayers = if ready then readyPlayers r - 1 else readyPlayers r |
|
3451
62089ccec75c
Uses StateT monad instead of manually maintaining the state
unc0rr
parents:
3436
diff
changeset
|
116 |
}) ri |
3566 | 117 |
|
3673
45778b16b224
Some comments on the reason of the bug, leave bug not fixed yet
unc0rr
parents:
3671
diff
changeset
|
118 |
processAction $ AnswerClients [chan] ["BYE", msg] |
3566 | 119 |
modify (\s -> s{removedClients = ci `Set.insert` removedClients s}) |
120 |
||
121 |
processAction (DeleteClient ci) = do |
|
3671
a94d1dc4a8d9
- burp's patch cleaning up module dependancies + cabal file
unc0rr
parents:
3658
diff
changeset
|
122 |
rnc <- gets roomsClients |
a94d1dc4a8d9
- burp's patch cleaning up module dependancies + cabal file
unc0rr
parents:
3658
diff
changeset
|
123 |
liftIO $ removeClient rnc ci |
3566 | 124 |
modify (\s -> s{removedClients = ci `Set.delete` removedClients s}) |
125 |
||
3436
288fcbdb77b6
Make server build again (it's still useless though)
unc0rr
parents:
3435
diff
changeset
|
126 |
{- |
2867
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
127 |
where |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
128 |
client = clients ! clID |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
129 |
clientNick = nick client |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
130 |
answerInformRoom = |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
131 |
if roomID client /= 0 then |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
132 |
if not $ Prelude.null msg then |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
133 |
[AnswerThisRoom ["LEFT", clientNick, msg]] |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
134 |
else |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
135 |
[AnswerThisRoom ["LEFT", clientNick]] |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
136 |
else |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
137 |
[] |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
138 |
answerOthersQuit = |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
139 |
if logonPassed client then |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
140 |
if not $ Prelude.null msg then |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
141 |
[AnswerAll ["LOBBY:LEFT", clientNick, msg]] |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
142 |
else |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
143 |
[AnswerAll ["LOBBY:LEFT", clientNick]] |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
144 |
else |
3436
288fcbdb77b6
Make server build again (it's still useless though)
unc0rr
parents:
3435
diff
changeset
|
145 |
[] |
288fcbdb77b6
Make server build again (it's still useless though)
unc0rr
parents:
3435
diff
changeset
|
146 |
-} |
1804 | 147 |
|
3458 | 148 |
processAction (ModifyClient f) = do |
149 |
(Just ci) <- gets clientIndex |
|
150 |
rnc <- gets roomsClients |
|
151 |
liftIO $ modifyClient rnc f ci |
|
152 |
return () |
|
3568 | 153 |
|
154 |
processAction (ModifyClient2 ci f) = do |
|
155 |
rnc <- gets roomsClients |
|
156 |
liftIO $ modifyClient rnc f ci |
|
157 |
return () |
|
158 |
||
1804 | 159 |
|
3458 | 160 |
processAction (ModifyRoom f) = do |
161 |
rnc <- gets roomsClients |
|
162 |
ri <- clientRoomA |
|
163 |
liftIO $ modifyRoom rnc f ri |
|
164 |
return () |
|
2403 | 165 |
|
3458 | 166 |
{- |
1804 | 167 |
|
3425 | 168 |
processAction (clID, serverInfo, rnc) (ModifyServerInfo func) = |
169 |
return (clID, func serverInfo, rnc) |
|
1925 | 170 |
|
3502 | 171 |
-} |
1925 | 172 |
|
3531 | 173 |
processAction (MoveToRoom ri) = do |
3502 | 174 |
(Just ci) <- gets clientIndex |
175 |
rnc <- gets roomsClients |
|
176 |
liftIO $ do |
|
177 |
modifyClient rnc (\cl -> cl{teamsInGame = 0}) ci |
|
3531 | 178 |
modifyRoom rnc (\r -> r{playersIn = (playersIn r) + 1}) ri |
179 |
||
180 |
liftIO $ moveClientToRoom rnc ri ci |
|
181 |
||
182 |
chans <- liftM (map sendChan) $ roomClientsS ri |
|
3502 | 183 |
clNick <- client's nick |
3645 | 184 |
|
3502 | 185 |
processAction $ AnswerClients chans ["JOINED", clNick] |
1804 | 186 |
|
3531 | 187 |
processAction (MoveToLobby msg) = do |
188 |
(Just ci) <- gets clientIndex |
|
189 |
--ri <- clientRoomA |
|
190 |
rnc <- gets roomsClients |
|
191 |
||
192 |
liftIO $ moveClientToLobby rnc ci |
|
193 |
||
3502 | 194 |
{- |
2867
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
195 |
(_, _, newClients, newRooms) <- |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
196 |
if isMaster client then |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
197 |
if (gameinprogress room) && (playersIn room > 1) then |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
198 |
(changeMaster >>= (\state -> foldM processAction state |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
199 |
[AnswerOthersInRoom ["LEFT", nick client, msg], |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
200 |
AnswerOthersInRoom ["WARNING", "Admin left the room"], |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
201 |
RemoveClientTeams clID])) |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
202 |
else -- not in game |
3425 | 203 |
processAction (clID, serverInfo, rnc) RemoveRoom |
2867
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
204 |
else -- not master |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
205 |
foldM |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
206 |
processAction |
3425 | 207 |
(clID, serverInfo, rnc) |
2867
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
208 |
[AnswerOthersInRoom ["LEFT", nick client, msg], |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
209 |
RemoveClientTeams clID] |
3531 | 210 |
|
3425 | 211 |
|
2867
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
212 |
return ( |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
213 |
clID, |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
214 |
serverInfo, |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
215 |
adjust resetClientFlags clID newClients, |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
216 |
adjust removeClientFromRoom rID $ adjust insertClientToRoom 0 newRooms |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
217 |
) |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
218 |
where |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
219 |
rID = roomID client |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
220 |
client = clients ! clID |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
221 |
room = rooms ! rID |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
222 |
resetClientFlags cl = cl{roomID = 0, isMaster = False, isReady = False, teamsInGame = undefined} |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
223 |
removeClientFromRoom r = r{ |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
224 |
playersIDs = otherPlayersSet, |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
225 |
playersIn = (playersIn r) - 1, |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
226 |
readyPlayers = if isReady client then (readyPlayers r) - 1 else readyPlayers r |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
227 |
} |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
228 |
insertClientToRoom r = r{playersIDs = IntSet.insert clID (playersIDs r)} |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
229 |
changeMaster = do |
3425 | 230 |
processAction (newMasterId, serverInfo, rnc) $ AnswerThisClient ["ROOM_CONTROL_ACCESS", "1"] |
2867
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
231 |
return ( |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
232 |
clID, |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
233 |
serverInfo, |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
234 |
adjust (\cl -> cl{isMaster = True}) newMasterId clients, |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
235 |
adjust (\r -> r{masterID = newMasterId, name = newRoomName}) rID rooms |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
236 |
) |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
237 |
newRoomName = nick newMasterClient |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
238 |
otherPlayersSet = IntSet.delete clID (playersIDs room) |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
239 |
newMasterId = IntSet.findMin otherPlayersSet |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
240 |
newMasterClient = clients ! newMasterId |
3502 | 241 |
-} |
1804 | 242 |
|
3502 | 243 |
processAction (AddRoom roomName roomPassword) = do |
3566 | 244 |
Just clId <- gets clientIndex |
245 |
rnc <- gets roomsClients |
|
3502 | 246 |
proto <- liftIO $ client'sM rnc clientProto clId |
3645 | 247 |
|
2867
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
248 |
let room = newRoom{ |
3502 | 249 |
masterID = clId, |
2867
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
250 |
name = roomName, |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
251 |
password = roomPassword, |
3502 | 252 |
roomProto = proto |
2867
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
253 |
} |
3645 | 254 |
|
255 |
rId <- liftIO $ addRoom rnc room |
|
256 |
||
257 |
processAction $ MoveToRoom rId |
|
258 |
||
3502 | 259 |
chans <- liftM (map sendChan) $ roomClientsS lobbyId |
1804 | 260 |
|
3502 | 261 |
mapM_ processAction [ |
262 |
AnswerClients chans ["ROOM", "ADD", roomName] |
|
263 |
, ModifyClient (\cl -> cl{isMaster = True}) |
|
3645 | 264 |
] |
1804 | 265 |
|
3502 | 266 |
{- |
3425 | 267 |
processAction (clID, serverInfo, rnc) (RemoveRoom) = do |
268 |
processAction (clID, serverInfo, rnc) $ AnswerLobby ["ROOM", "DEL", name room] |
|
269 |
processAction (clID, serverInfo, rnc) $ AnswerOthersInRoom ["ROOMABANDONED", name room] |
|
2867
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
270 |
return (clID, |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
271 |
serverInfo, |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
272 |
Data.IntMap.map (\cl -> if roomID cl == rID then cl{roomID = 0, isMaster = False, isReady = False, teamsInGame = undefined} else cl) clients, |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
273 |
delete rID $ adjust (\r -> r{playersIDs = IntSet.union (playersIDs room) (playersIDs r)}) 0 rooms |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
274 |
) |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
275 |
where |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
276 |
room = rooms ! rID |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
277 |
rID = roomID client |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
278 |
client = clients ! clID |
1804 | 279 |
|
3656 | 280 |
-} |
281 |
processAction (UnreadyRoomClients) = do |
|
282 |
rnc <- gets roomsClients |
|
283 |
ri <- clientRoomA |
|
284 |
roomPlayers <- roomClientsS ri |
|
285 |
roomClIDs <- liftIO $ roomClientsIndicesM rnc ri |
|
286 |
processAction $ AnswerClients (map sendChan roomPlayers) ("NOT_READY" : map nick roomPlayers) |
|
287 |
liftIO $ mapM_ (modifyClient rnc (\cl -> cl{isReady = False})) roomClIDs |
|
288 |
processAction $ ModifyRoom (\r -> r{readyPlayers = 0}) |
|
1813 | 289 |
|
1811 | 290 |
|
3645 | 291 |
processAction (RemoveTeam teamName) = do |
292 |
rnc <- gets roomsClients |
|
293 |
cl <- client's id |
|
294 |
ri <- clientRoomA |
|
295 |
inGame <- liftIO $ room'sM rnc gameinprogress ri |
|
296 |
chans <- liftM (map sendChan . filter (/= cl)) $ roomClientsS ri |
|
297 |
if inGame then |
|
298 |
mapM_ processAction [ |
|
299 |
AnswerClients chans ["REMOVE_TEAM", teamName], |
|
300 |
ModifyRoom (\r -> r{teams = Prelude.filter (\t -> teamName /= teamname t) $ teams r}) |
|
301 |
] |
|
2867
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
302 |
else |
3645 | 303 |
mapM_ processAction [ |
304 |
AnswerClients chans ["EM", rmTeamMsg], |
|
305 |
ModifyRoom (\r -> r{ |
|
306 |
teams = Prelude.filter (\t -> teamName /= teamname t) $ teams r, |
|
307 |
leftTeams = teamName : leftTeams r, |
|
308 |
roundMsgs = roundMsgs r Seq.|> rmTeamMsg |
|
309 |
}) |
|
310 |
] |
|
2867
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
311 |
where |
3645 | 312 |
rmTeamMsg = toEngineMsg $ (B.singleton 'F') `B.append` teamName |
1813 | 313 |
|
3458 | 314 |
processAction CheckRegistered = do |
315 |
(Just ci) <- gets clientIndex |
|
3501 | 316 |
n <- client's nick |
317 |
h <- client's host |
|
3458 | 318 |
db <- gets (dbQueries . serverInfo) |
319 |
liftIO $ writeChan db $ CheckAccount ci n h |
|
320 |
return () |
|
1834 | 321 |
|
3458 | 322 |
{- |
3425 | 323 |
processAction (clID, serverInfo, rnc) (ClearAccountsCache) = do |
2867
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
324 |
writeChan (dbQueries serverInfo) ClearCache |
3425 | 325 |
return (clID, serverInfo, rnc) |
2867
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
326 |
where |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
327 |
client = clients ! clID |
2155
d897222d3339
Implement ability for server admin to clear accounts cache
unc0rr
parents:
2126
diff
changeset
|
328 |
|
d897222d3339
Implement ability for server admin to clear accounts cache
unc0rr
parents:
2126
diff
changeset
|
329 |
|
3425 | 330 |
processAction (clID, serverInfo, rnc) (Dump) = do |
2867
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
331 |
writeChan (sendChan $ clients ! clID) ["DUMP", show serverInfo, showTree clients, showTree rooms] |
3425 | 332 |
return (clID, serverInfo, rnc) |
3501 | 333 |
-} |
1841
fba7210b438b
Retrieve client password from web database and ask for it
unc0rr
parents:
1839
diff
changeset
|
334 |
|
3501 | 335 |
processAction (ProcessAccountInfo info) = |
2867
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
336 |
case info of |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
337 |
HasAccount passwd isAdmin -> do |
3501 | 338 |
chan <- client's sendChan |
3673
45778b16b224
Some comments on the reason of the bug, leave bug not fixed yet
unc0rr
parents:
3671
diff
changeset
|
339 |
processAction $ AnswerClients [chan] ["ASKPASSWORD"] |
2867
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
340 |
Guest -> do |
3502 | 341 |
processAction JoinLobby |
2867
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
342 |
Admin -> do |
3502 | 343 |
mapM processAction [ModifyClient (\cl -> cl{isAdministrator = True}), JoinLobby] |
3501 | 344 |
chan <- client's sendChan |
3673
45778b16b224
Some comments on the reason of the bug, leave bug not fixed yet
unc0rr
parents:
3671
diff
changeset
|
345 |
processAction $ AnswerClients [chan] ["ADMIN_ACCESS"] |
1841
fba7210b438b
Retrieve client password from web database and ask for it
unc0rr
parents:
1839
diff
changeset
|
346 |
|
3502 | 347 |
|
348 |
processAction JoinLobby = do |
|
3501 | 349 |
chan <- client's sendChan |
3502 | 350 |
clientNick <- client's nick |
3566 | 351 |
(lobbyNicks, clientsChans) <- liftM (unzip . Prelude.map (\c -> (nick c, sendChan c)) . Prelude.filter logonPassed) $! allClientsS |
3501 | 352 |
mapM_ processAction $ |
3502 | 353 |
(AnswerClients clientsChans ["LOBBY:JOINED", clientNick]) |
3566 | 354 |
: [AnswerClients [chan] ("LOBBY:JOINED" : clientNick : lobbyNicks)] |
3502 | 355 |
++ [ModifyClient (\cl -> cl{logonPassed = True}), SendServerMessage] |
2118
0ebcc98ebc1a
Send server message after nicks info (more chance for it to be seen)
unc0rr
parents:
2116
diff
changeset
|
356 |
|
3501 | 357 |
{- |
3502 | 358 |
processAction (clID, serverInfo, rnc) (RoomAddThisClient rID) = |
359 |
processAction ( |
|
360 |
clID, |
|
361 |
serverInfo, |
|
362 |
adjust (\cl -> cl{roomID = rID, teamsInGame = if rID == 0 then teamsInGame cl else 0}) clID clients, |
|
363 |
adjust (\r -> r{playersIDs = IntSet.insert clID (playersIDs r), playersIn = (playersIn r) + 1}) rID $ |
|
364 |
adjust (\r -> r{playersIDs = IntSet.delete clID (playersIDs r)}) 0 rooms |
|
365 |
) joinMsg |
|
366 |
where |
|
367 |
client = clients ! clID |
|
368 |
joinMsg = if rID == 0 then |
|
369 |
AnswerAllOthers ["LOBBY:JOINED", nick client] |
|
370 |
else |
|
371 |
AnswerThisRoom ["JOINED", nick client] |
|
1841
fba7210b438b
Retrieve client password from web database and ask for it
unc0rr
parents:
1839
diff
changeset
|
372 |
|
3425 | 373 |
processAction (clID, serverInfo, rnc) (KickClient kickID) = |
374 |
liftM2 replaceID (return clID) (processAction (kickID, serverInfo, rnc) $ ByeClient "Kicked") |
|
1879 | 375 |
|
1926
cb46fbdcaa41
Add simple DoS protection mechanism (although better than previous server had)
unc0rr
parents:
1925
diff
changeset
|
376 |
|
3425 | 377 |
processAction (clID, serverInfo, rnc) (BanClient banNick) = |
378 |
return (clID, serverInfo, rnc) |
|
1921 | 379 |
|
1879 | 380 |
|
3425 | 381 |
processAction (clID, serverInfo, rnc) (KickRoomClient kickID) = do |
2867
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
382 |
writeChan (sendChan $ clients ! kickID) ["KICKED"] |
3425 | 383 |
liftM2 replaceID (return clID) (processAction (kickID, serverInfo, rnc) $ RoomRemoveThisClient "kicked") |
1879 | 384 |
|
385 |
||
3425 | 386 |
processAction (clID, serverInfo, rnc) (RemoveClientTeams teamsClID) = |
2867
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
387 |
liftM2 replaceID (return clID) $ |
3425 | 388 |
foldM processAction (teamsClID, serverInfo, rnc) removeTeamsActions |
2867
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
389 |
where |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
390 |
client = clients ! teamsClID |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
391 |
room = rooms ! (roomID client) |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
392 |
teamsToRemove = Prelude.filter (\t -> teamowner t == nick client) $ teams room |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
393 |
removeTeamsActions = Prelude.map (RemoveTeam . teamname) teamsToRemove |
3435 | 394 |
-} |
1926
cb46fbdcaa41
Add simple DoS protection mechanism (although better than previous server had)
unc0rr
parents:
1925
diff
changeset
|
395 |
|
3451
62089ccec75c
Uses StateT monad instead of manually maintaining the state
unc0rr
parents:
3436
diff
changeset
|
396 |
processAction (AddClient client) = do |
62089ccec75c
Uses StateT monad instead of manually maintaining the state
unc0rr
parents:
3436
diff
changeset
|
397 |
rnc <- gets roomsClients |
62089ccec75c
Uses StateT monad instead of manually maintaining the state
unc0rr
parents:
3436
diff
changeset
|
398 |
si <- gets serverInfo |
62089ccec75c
Uses StateT monad instead of manually maintaining the state
unc0rr
parents:
3436
diff
changeset
|
399 |
liftIO $ do |
62089ccec75c
Uses StateT monad instead of manually maintaining the state
unc0rr
parents:
3436
diff
changeset
|
400 |
ci <- addClient rnc client |
3500
af8390d807d6
Use sockets instead of handles, use bytestrings instead of strings
unc0rr
parents:
3458
diff
changeset
|
401 |
forkIO $ clientRecvLoop (clientSocket client) (coreChan si) ci |
af8390d807d6
Use sockets instead of handles, use bytestrings instead of strings
unc0rr
parents:
3458
diff
changeset
|
402 |
forkIO $ clientSendLoop (clientSocket client) (coreChan si) (sendChan client) ci |
3451
62089ccec75c
Uses StateT monad instead of manually maintaining the state
unc0rr
parents:
3436
diff
changeset
|
403 |
|
62089ccec75c
Uses StateT monad instead of manually maintaining the state
unc0rr
parents:
3436
diff
changeset
|
404 |
infoM "Clients" (show ci ++ ": New client. Time: " ++ show (connectTime client)) |
3425 | 405 |
|
3673
45778b16b224
Some comments on the reason of the bug, leave bug not fixed yet
unc0rr
parents:
3671
diff
changeset
|
406 |
processAction $ AnswerClients [sendChan client] ["CONNECTED", "Hedgewars server http://www.hedgewars.org/"] |
3451
62089ccec75c
Uses StateT monad instead of manually maintaining the state
unc0rr
parents:
3436
diff
changeset
|
407 |
{- let newLogins = takeWhile (\(_ , time) -> (connectTime client) `diffUTCTime` time <= 11) $ lastLogins serverInfo |
1926
cb46fbdcaa41
Add simple DoS protection mechanism (although better than previous server had)
unc0rr
parents:
1925
diff
changeset
|
408 |
|
3451
62089ccec75c
Uses StateT monad instead of manually maintaining the state
unc0rr
parents:
3436
diff
changeset
|
409 |
if False && (isJust $ host client `Prelude.lookup` newLogins) then |
62089ccec75c
Uses StateT monad instead of manually maintaining the state
unc0rr
parents:
3436
diff
changeset
|
410 |
processAction (ci, serverInfo{lastLogins = newLogins}, rnc) $ ByeClient "Reconnected too fast" |
62089ccec75c
Uses StateT monad instead of manually maintaining the state
unc0rr
parents:
3436
diff
changeset
|
411 |
else |
62089ccec75c
Uses StateT monad instead of manually maintaining the state
unc0rr
parents:
3436
diff
changeset
|
412 |
return (ci, serverInfo) |
62089ccec75c
Uses StateT monad instead of manually maintaining the state
unc0rr
parents:
3436
diff
changeset
|
413 |
-} |
62089ccec75c
Uses StateT monad instead of manually maintaining the state
unc0rr
parents:
3436
diff
changeset
|
414 |
|
62089ccec75c
Uses StateT monad instead of manually maintaining the state
unc0rr
parents:
3436
diff
changeset
|
415 |
|
1927
e2031906a347
Ping clients every 30 seconds. Disconnection due to ping timeout to be implemented.
unc0rr
parents:
1926
diff
changeset
|
416 |
|
3654 | 417 |
processAction PingAll = do |
418 |
rnc <- gets roomsClients |
|
3658
113cb9345be1
Make server stay alive when some clients get kicked
unc0rr
parents:
3656
diff
changeset
|
419 |
liftIO (allClientsM rnc) >>= mapM_ (kickTimeouted rnc) |
3654 | 420 |
cis <- liftIO $ allClientsM rnc |
421 |
chans <- liftIO $ mapM (client'sM rnc sendChan) cis |
|
422 |
liftIO $ mapM_ (modifyClient rnc (\cl -> cl{pingsQueue = pingsQueue cl + 1})) cis |
|
423 |
processAction $ AnswerClients chans ["PING"] |
|
2867
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
424 |
where |
3654 | 425 |
kickTimeouted rnc ci = do |
426 |
pq <- liftIO $ client'sM rnc pingsQueue ci |
|
427 |
when (pq > 0) $ |
|
428 |
withStateT (\as -> as{clientIndex = Just ci}) $ |
|
429 |
processAction (ByeClient "Ping timeout") |
|
2172 | 430 |
|
431 |
||
3653 | 432 |
processAction (StatsAction) = do |
433 |
rnc <- gets roomsClients |
|
434 |
si <- gets serverInfo |
|
435 |
(roomsNum, clientsNum) <- liftIO $ withRoomsAndClients rnc stats |
|
436 |
liftIO $ writeChan (dbQueries si) $ SendStats clientsNum (roomsNum - 1) |
|
437 |
where |
|
438 |
stats irnc = (length $ allRooms irnc, length $ allClients irnc) |
|
3425 | 439 |