author | unc0rr |
Sat, 20 Jun 2009 07:50:03 +0000 | |
changeset 2176 | fc082baf448c |
parent 2172 | 80d34c0b9dfe |
child 2245 | c011aecc95e5 |
permissions | -rw-r--r-- |
1804 | 1 |
module Actions where |
2 |
||
3 |
import Control.Concurrent.STM |
|
4 |
import Control.Concurrent.Chan |
|
5 |
import Data.IntMap |
|
6 |
import qualified Data.IntSet as IntSet |
|
1813 | 7 |
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
|
8 |
import System.Log.Logger |
1804 | 9 |
import Monad |
1926
cb46fbdcaa41
Add simple DoS protection mechanism (although better than previous server had)
unc0rr
parents:
1925
diff
changeset
|
10 |
import Data.Time |
cb46fbdcaa41
Add simple DoS protection mechanism (although better than previous server had)
unc0rr
parents:
1925
diff
changeset
|
11 |
import Maybe |
1804 | 12 |
----------------------------- |
13 |
import CoreTypes |
|
1813 | 14 |
import Utils |
1804 | 15 |
|
16 |
data Action = |
|
17 |
AnswerThisClient [String] |
|
18 |
| AnswerAll [String] |
|
19 |
| AnswerAllOthers [String] |
|
20 |
| AnswerThisRoom [String] |
|
21 |
| AnswerOthersInRoom [String] |
|
22 |
| AnswerLobby [String] |
|
1923 | 23 |
| SendServerMessage |
1804 | 24 |
| RoomAddThisClient Int -- roomID |
2126 | 25 |
| RoomRemoveThisClient String |
1813 | 26 |
| RemoveTeam String |
1804 | 27 |
| RemoveRoom |
1811 | 28 |
| UnreadyRoomClients |
1841
fba7210b438b
Retrieve client password from web database and ask for it
unc0rr
parents:
1839
diff
changeset
|
29 |
| MoveToLobby |
1804 | 30 |
| ProtocolError String |
31 |
| Warning String |
|
32 |
| ByeClient String |
|
1862 | 33 |
| KickClient Int -- clID |
1879 | 34 |
| KickRoomClient Int -- clID |
1921 | 35 |
| BanClient String -- nick |
1879 | 36 |
| RemoveClientTeams Int -- clID |
1804 | 37 |
| ModifyClient (ClientInfo -> ClientInfo) |
38 |
| ModifyRoom (RoomInfo -> RoomInfo) |
|
1925 | 39 |
| ModifyServerInfo (ServerInfo -> ServerInfo) |
1804 | 40 |
| AddRoom String String |
1834 | 41 |
| CheckRegistered |
2155
d897222d3339
Implement ability for server admin to clear accounts cache
unc0rr
parents:
2126
diff
changeset
|
42 |
| ClearAccountsCache |
1839
5dd4cb7fd7e5
Server now send ASKPASSWORD command to frontend when user has web account
unc0rr
parents:
1834
diff
changeset
|
43 |
| ProcessAccountInfo AccountInfo |
1804 | 44 |
| Dump |
1926
cb46fbdcaa41
Add simple DoS protection mechanism (although better than previous server had)
unc0rr
parents:
1925
diff
changeset
|
45 |
| AddClient ClientInfo |
1927
e2031906a347
Ping clients every 30 seconds. Disconnection due to ping timeout to be implemented.
unc0rr
parents:
1926
diff
changeset
|
46 |
| PingAll |
2172 | 47 |
| StatsAction |
1804 | 48 |
|
49 |
type CmdHandler = Int -> Clients -> Rooms -> [String] -> [Action] |
|
50 |
||
1862 | 51 |
replaceID a (b, c, d, e) = (a, c, d, e) |
1804 | 52 |
|
53 |
processAction :: (Int, ServerInfo, Clients, Rooms) -> Action -> IO (Int, ServerInfo, Clients, Rooms) |
|
54 |
||
55 |
||
56 |
processAction (clID, serverInfo, clients, rooms) (AnswerThisClient msg) = do |
|
57 |
writeChan (sendChan $ clients ! clID) msg |
|
58 |
return (clID, serverInfo, clients, rooms) |
|
59 |
||
60 |
||
61 |
processAction (clID, serverInfo, clients, rooms) (AnswerAll msg) = do |
|
1927
e2031906a347
Ping clients every 30 seconds. Disconnection due to ping timeout to be implemented.
unc0rr
parents:
1926
diff
changeset
|
62 |
mapM_ (\cl -> writeChan (sendChan cl) msg) (elems clients) |
1804 | 63 |
return (clID, serverInfo, clients, rooms) |
64 |
||
65 |
||
66 |
processAction (clID, serverInfo, clients, rooms) (AnswerAllOthers msg) = do |
|
1879 | 67 |
mapM_ (\id -> writeChan (sendChan $ clients ! id) msg) $ |
68 |
Prelude.filter (\id' -> (id' /= clID) && (logonPassed $ clients ! id')) (keys clients) |
|
1804 | 69 |
return (clID, serverInfo, clients, rooms) |
70 |
||
71 |
||
72 |
processAction (clID, serverInfo, clients, rooms) (AnswerThisRoom msg) = do |
|
73 |
mapM_ (\id -> writeChan (sendChan $ clients ! id) msg) roomClients |
|
74 |
return (clID, serverInfo, clients, rooms) |
|
75 |
where |
|
76 |
roomClients = IntSet.elems $ playersIDs room |
|
77 |
room = rooms ! rID |
|
78 |
rID = roomID client |
|
79 |
client = clients ! clID |
|
80 |
||
81 |
||
82 |
processAction (clID, serverInfo, clients, rooms) (AnswerOthersInRoom msg) = do |
|
83 |
mapM_ (\id -> writeChan (sendChan $ clients ! id) msg) $ Prelude.filter (/= clID) roomClients |
|
84 |
return (clID, serverInfo, clients, rooms) |
|
85 |
where |
|
86 |
roomClients = IntSet.elems $ playersIDs room |
|
87 |
room = rooms ! rID |
|
88 |
rID = roomID client |
|
89 |
client = clients ! clID |
|
90 |
||
91 |
||
92 |
processAction (clID, serverInfo, clients, rooms) (AnswerLobby msg) = do |
|
93 |
mapM_ (\id -> writeChan (sendChan $ clients ! id) msg) roomClients |
|
94 |
return (clID, serverInfo, clients, rooms) |
|
95 |
where |
|
96 |
roomClients = IntSet.elems $ playersIDs room |
|
97 |
room = rooms ! 0 |
|
98 |
||
99 |
||
1923 | 100 |
processAction (clID, serverInfo, clients, rooms) SendServerMessage = do |
1953 | 101 |
writeChan (sendChan $ clients ! clID) $ ["SERVER_MESSAGE", message serverInfo] |
1923 | 102 |
return (clID, serverInfo, clients, rooms) |
1953 | 103 |
where |
104 |
client = clients ! clID |
|
2104 | 105 |
message = if clientProto client < 27 then |
1953 | 106 |
serverMessageForOldVersions |
107 |
else |
|
1994 | 108 |
serverMessage |
1923 | 109 |
|
110 |
||
1804 | 111 |
processAction (clID, serverInfo, clients, rooms) (ProtocolError msg) = do |
112 |
writeChan (sendChan $ clients ! clID) ["ERROR", msg] |
|
113 |
return (clID, serverInfo, clients, rooms) |
|
114 |
||
115 |
||
116 |
processAction (clID, serverInfo, clients, rooms) (Warning msg) = do |
|
117 |
writeChan (sendChan $ clients ! clID) ["WARNING", msg] |
|
118 |
return (clID, serverInfo, clients, rooms) |
|
119 |
||
120 |
||
121 |
processAction (clID, serverInfo, clients, rooms) (ByeClient msg) = do |
|
2001 | 122 |
infoM "Clients" ((show $ clientUID client) ++ " quits: " ++ msg) |
1929 | 123 |
(_, _, newClients, newRooms) <- |
1930 | 124 |
if roomID client /= 0 then |
125 |
processAction (clID, serverInfo, clients, rooms) |
|
1929 | 126 |
(if isMaster client then RemoveRoom else RemoveClientTeams clID) |
1930 | 127 |
else |
128 |
return (clID, serverInfo, clients, rooms) |
|
1929 | 129 |
|
130 |
mapM_ (processAction (clID, serverInfo, newClients, newRooms)) $ answerOthersQuit ++ answerInformRoom |
|
1926
cb46fbdcaa41
Add simple DoS protection mechanism (although better than previous server had)
unc0rr
parents:
1925
diff
changeset
|
131 |
writeChan (sendChan $ clients ! clID) ["BYE", msg] |
1804 | 132 |
return ( |
133 |
0, |
|
134 |
serverInfo, |
|
1929 | 135 |
delete clID newClients, |
1823 | 136 |
adjust (\r -> r{ |
137 |
playersIDs = IntSet.delete clID (playersIDs r), |
|
138 |
playersIn = (playersIn r) - 1, |
|
139 |
readyPlayers = if isReady client then readyPlayers r - 1 else readyPlayers r |
|
1930 | 140 |
}) (roomID $ newClients ! clID) newRooms |
1804 | 141 |
) |
142 |
where |
|
143 |
client = clients ! clID |
|
144 |
clientNick = nick client |
|
145 |
answerInformRoom = |
|
146 |
if roomID client /= 0 then |
|
147 |
if not $ Prelude.null msg then |
|
148 |
[AnswerThisRoom ["LEFT", clientNick, msg]] |
|
149 |
else |
|
150 |
[AnswerThisRoom ["LEFT", clientNick]] |
|
151 |
else |
|
152 |
[] |
|
153 |
answerOthersQuit = |
|
1846 | 154 |
if logonPassed client then |
1804 | 155 |
if not $ Prelude.null msg then |
156 |
[AnswerAll ["LOBBY:LEFT", clientNick, msg]] |
|
157 |
else |
|
158 |
[AnswerAll ["LOBBY:LEFT", clientNick]] |
|
159 |
else |
|
160 |
[] |
|
161 |
||
162 |
||
163 |
processAction (clID, serverInfo, clients, rooms) (ModifyClient func) = do |
|
164 |
return (clID, serverInfo, adjust func clID clients, rooms) |
|
165 |
||
166 |
||
167 |
processAction (clID, serverInfo, clients, rooms) (ModifyRoom func) = do |
|
168 |
return (clID, serverInfo, clients, adjust func rID rooms) |
|
169 |
where |
|
170 |
rID = roomID $ clients ! clID |
|
171 |
||
172 |
||
1925 | 173 |
processAction (clID, serverInfo, clients, rooms) (ModifyServerInfo func) = do |
174 |
return (clID, func serverInfo, clients, rooms) |
|
175 |
||
176 |
||
1804 | 177 |
processAction (clID, serverInfo, clients, rooms) (RoomAddThisClient rID) = do |
178 |
processAction ( |
|
179 |
clID, |
|
180 |
serverInfo, |
|
181 |
adjust (\cl -> cl{roomID = rID}) clID clients, |
|
182 |
adjust (\r -> r{playersIDs = IntSet.insert clID (playersIDs r), playersIn = (playersIn r) + 1}) rID $ |
|
183 |
adjust (\r -> r{playersIDs = IntSet.delete clID (playersIDs r)}) 0 rooms |
|
184 |
) joinMsg |
|
185 |
where |
|
186 |
client = clients ! clID |
|
187 |
joinMsg = if rID == 0 then |
|
188 |
AnswerAllOthers ["LOBBY:JOINED", nick client] |
|
189 |
else |
|
190 |
AnswerThisRoom ["JOINED", nick client] |
|
191 |
||
192 |
||
2126 | 193 |
processAction (clID, serverInfo, clients, rooms) (RoomRemoveThisClient msg) = do |
1977 | 194 |
(_, _, newClients, newRooms) <- |
195 |
if roomID client /= 0 then |
|
196 |
foldM |
|
197 |
processAction |
|
198 |
(clID, serverInfo, clients, rooms) |
|
2126 | 199 |
[AnswerOthersInRoom ["LEFT", nick client, msg], |
1977 | 200 |
RemoveClientTeams clID] |
201 |
else |
|
202 |
return (clID, serverInfo, clients, rooms) |
|
203 |
||
1804 | 204 |
return ( |
205 |
clID, |
|
206 |
serverInfo, |
|
1977 | 207 |
adjust (\cl -> cl{roomID = 0, isMaster = False, isReady = False}) clID newClients, |
1823 | 208 |
adjust (\r -> r{ |
209 |
playersIDs = IntSet.delete clID (playersIDs r), |
|
210 |
playersIn = (playersIn r) - 1, |
|
1827 | 211 |
readyPlayers = if isReady client then (readyPlayers r) - 1 else readyPlayers r |
1823 | 212 |
}) rID $ |
1977 | 213 |
adjust (\r -> r{playersIDs = IntSet.insert clID (playersIDs r)}) 0 newRooms |
1804 | 214 |
) |
215 |
where |
|
216 |
rID = roomID client |
|
217 |
client = clients ! clID |
|
218 |
||
219 |
||
220 |
processAction (clID, serverInfo, clients, rooms) (AddRoom roomName roomPassword) = do |
|
221 |
let newServerInfo = serverInfo {nextRoomID = newID} |
|
222 |
let room = newRoom{ |
|
223 |
roomUID = newID, |
|
224 |
name = roomName, |
|
225 |
password = roomPassword, |
|
226 |
roomProto = (clientProto client) |
|
227 |
} |
|
228 |
||
229 |
processAction (clID, serverInfo, clients, rooms) $ AnswerLobby ["ROOM", "ADD", roomName] |
|
230 |
||
231 |
processAction ( |
|
232 |
clID, |
|
233 |
newServerInfo, |
|
234 |
adjust (\cl -> cl{isMaster = True}) clID clients, |
|
235 |
insert newID room rooms |
|
236 |
) $ RoomAddThisClient newID |
|
237 |
where |
|
238 |
newID = (nextRoomID serverInfo) - 1 |
|
239 |
client = clients ! clID |
|
240 |
||
241 |
||
242 |
processAction (clID, serverInfo, clients, rooms) (RemoveRoom) = do |
|
1811 | 243 |
processAction (clID, serverInfo, clients, rooms) $ AnswerLobby ["ROOM", "DEL", name room] |
244 |
processAction (clID, serverInfo, clients, rooms) $ AnswerOthersInRoom ["ROOMABANDONED", name room] |
|
1804 | 245 |
return (clID, |
246 |
serverInfo, |
|
1827 | 247 |
Data.IntMap.map (\cl -> if roomID cl == rID then cl{roomID = 0, isMaster = False, isReady = False} else cl) clients, |
1811 | 248 |
delete rID $ adjust (\r -> r{playersIDs = IntSet.union (playersIDs room) (playersIDs r)}) 0 rooms |
1804 | 249 |
) |
250 |
where |
|
1811 | 251 |
room = rooms ! rID |
1804 | 252 |
rID = roomID client |
253 |
client = clients ! clID |
|
254 |
||
1813 | 255 |
|
1811 | 256 |
processAction (clID, serverInfo, clients, rooms) (UnreadyRoomClients) = do |
257 |
processAction (clID, serverInfo, clients, rooms) $ AnswerThisRoom ("NOT_READY" : roomPlayers) |
|
258 |
return (clID, |
|
259 |
serverInfo, |
|
260 |
Data.IntMap.map (\cl -> if roomID cl == rID then cl{isReady = False} else cl) clients, |
|
1827 | 261 |
adjust (\r -> r{readyPlayers = 0}) rID rooms) |
1811 | 262 |
where |
263 |
room = rooms ! rID |
|
264 |
rID = roomID client |
|
265 |
client = clients ! clID |
|
266 |
roomPlayers = Prelude.map (nick . (clients !)) roomPlayersIDs |
|
267 |
roomPlayersIDs = IntSet.elems $ playersIDs room |
|
268 |
||
269 |
||
1813 | 270 |
processAction (clID, serverInfo, clients, rooms) (RemoveTeam teamName) = do |
271 |
newRooms <- if not $ gameinprogress room then |
|
272 |
do |
|
273 |
processAction (clID, serverInfo, clients, rooms) $ AnswerOthersInRoom ["REMOVE_TEAM", teamName] |
|
274 |
return $ |
|
275 |
adjust (\r -> r{teams = Prelude.filter (\t -> teamName /= teamname t) $ teams r}) rID rooms |
|
276 |
else |
|
277 |
do |
|
1866 | 278 |
processAction (clID, serverInfo, clients, rooms) $ AnswerOthersInRoom ["EM", rmTeamMsg] |
1813 | 279 |
return $ |
280 |
adjust (\r -> r{ |
|
281 |
teams = Prelude.filter (\t -> teamName /= teamname t) $ teams r, |
|
282 |
leftTeams = teamName : leftTeams r, |
|
283 |
roundMsgs = roundMsgs r Seq.|> rmTeamMsg |
|
284 |
}) rID rooms |
|
285 |
return (clID, serverInfo, clients, newRooms) |
|
286 |
where |
|
287 |
room = rooms ! rID |
|
288 |
rID = roomID client |
|
289 |
client = clients ! clID |
|
290 |
rmTeamMsg = toEngineMsg $ 'F' : teamName |
|
291 |
||
292 |
||
1834 | 293 |
processAction (clID, serverInfo, clients, rooms) (CheckRegistered) = do |
2116
dec7ead2d178
Bring back authentication to official server, now using separate process to perform database interaction
unc0rr
parents:
2104
diff
changeset
|
294 |
writeChan (dbQueries serverInfo) $ CheckAccount (clientUID client) (nick client) (host client) |
1834 | 295 |
return (clID, serverInfo, clients, rooms) |
296 |
where |
|
297 |
client = clients ! clID |
|
298 |
||
1841
fba7210b438b
Retrieve client password from web database and ask for it
unc0rr
parents:
1839
diff
changeset
|
299 |
|
2155
d897222d3339
Implement ability for server admin to clear accounts cache
unc0rr
parents:
2126
diff
changeset
|
300 |
processAction (clID, serverInfo, clients, rooms) (ClearAccountsCache) = do |
d897222d3339
Implement ability for server admin to clear accounts cache
unc0rr
parents:
2126
diff
changeset
|
301 |
writeChan (dbQueries serverInfo) $ ClearCache |
d897222d3339
Implement ability for server admin to clear accounts cache
unc0rr
parents:
2126
diff
changeset
|
302 |
return (clID, serverInfo, clients, rooms) |
d897222d3339
Implement ability for server admin to clear accounts cache
unc0rr
parents:
2126
diff
changeset
|
303 |
where |
d897222d3339
Implement ability for server admin to clear accounts cache
unc0rr
parents:
2126
diff
changeset
|
304 |
client = clients ! clID |
d897222d3339
Implement ability for server admin to clear accounts cache
unc0rr
parents:
2126
diff
changeset
|
305 |
|
d897222d3339
Implement ability for server admin to clear accounts cache
unc0rr
parents:
2126
diff
changeset
|
306 |
|
1804 | 307 |
processAction (clID, serverInfo, clients, rooms) (Dump) = do |
308 |
writeChan (sendChan $ clients ! clID) ["DUMP", show serverInfo, showTree clients, showTree rooms] |
|
309 |
return (clID, serverInfo, clients, rooms) |
|
310 |
||
1841
fba7210b438b
Retrieve client password from web database and ask for it
unc0rr
parents:
1839
diff
changeset
|
311 |
|
1839
5dd4cb7fd7e5
Server now send ASKPASSWORD command to frontend when user has web account
unc0rr
parents:
1834
diff
changeset
|
312 |
processAction (clID, serverInfo, clients, rooms) (ProcessAccountInfo info) = do |
5dd4cb7fd7e5
Server now send ASKPASSWORD command to frontend when user has web account
unc0rr
parents:
1834
diff
changeset
|
313 |
case info of |
1847
2178c0fc838c
Set admin flag and send admin notification to users with rid equal to 3
unc0rr
parents:
1846
diff
changeset
|
314 |
HasAccount passwd isAdmin -> do |
1839
5dd4cb7fd7e5
Server now send ASKPASSWORD command to frontend when user has web account
unc0rr
parents:
1834
diff
changeset
|
315 |
infoM "Clients" $ show clID ++ " has account" |
5dd4cb7fd7e5
Server now send ASKPASSWORD command to frontend when user has web account
unc0rr
parents:
1834
diff
changeset
|
316 |
writeChan (sendChan $ clients ! clID) ["ASKPASSWORD"] |
1847
2178c0fc838c
Set admin flag and send admin notification to users with rid equal to 3
unc0rr
parents:
1846
diff
changeset
|
317 |
return (clID, serverInfo, adjust (\cl -> cl{webPassword = passwd, isAdministrator = isAdmin}) clID clients, rooms) |
1839
5dd4cb7fd7e5
Server now send ASKPASSWORD command to frontend when user has web account
unc0rr
parents:
1834
diff
changeset
|
318 |
Guest -> do |
5dd4cb7fd7e5
Server now send ASKPASSWORD command to frontend when user has web account
unc0rr
parents:
1834
diff
changeset
|
319 |
infoM "Clients" $ show clID ++ " is guest" |
1841
fba7210b438b
Retrieve client password from web database and ask for it
unc0rr
parents:
1839
diff
changeset
|
320 |
processAction (clID, serverInfo, adjust (\cl -> cl{logonPassed = True}) clID clients, rooms) MoveToLobby |
1921 | 321 |
Admin -> do |
322 |
infoM "Clients" $ show clID ++ " is admin" |
|
323 |
foldM processAction (clID, serverInfo, adjust (\cl -> cl{logonPassed = True, isAdministrator = True}) clID clients, rooms) [MoveToLobby, AnswerThisClient ["ADMIN_ACCESS"]] |
|
1841
fba7210b438b
Retrieve client password from web database and ask for it
unc0rr
parents:
1839
diff
changeset
|
324 |
|
fba7210b438b
Retrieve client password from web database and ask for it
unc0rr
parents:
1839
diff
changeset
|
325 |
|
fba7210b438b
Retrieve client password from web database and ask for it
unc0rr
parents:
1839
diff
changeset
|
326 |
processAction (clID, serverInfo, clients, rooms) (MoveToLobby) = do |
fba7210b438b
Retrieve client password from web database and ask for it
unc0rr
parents:
1839
diff
changeset
|
327 |
foldM processAction (clID, serverInfo, clients, rooms) $ |
fba7210b438b
Retrieve client password from web database and ask for it
unc0rr
parents:
1839
diff
changeset
|
328 |
(RoomAddThisClient 0) |
fba7210b438b
Retrieve client password from web database and ask for it
unc0rr
parents:
1839
diff
changeset
|
329 |
: answerLobbyNicks |
2118
0ebcc98ebc1a
Send server message after nicks info (more chance for it to be seen)
unc0rr
parents:
2116
diff
changeset
|
330 |
++ [SendServerMessage] |
0ebcc98ebc1a
Send server message after nicks info (more chance for it to be seen)
unc0rr
parents:
2116
diff
changeset
|
331 |
|
1841
fba7210b438b
Retrieve client password from web database and ask for it
unc0rr
parents:
1839
diff
changeset
|
332 |
-- ++ (answerServerMessage client clients) |
fba7210b438b
Retrieve client password from web database and ask for it
unc0rr
parents:
1839
diff
changeset
|
333 |
where |
1846 | 334 |
lobbyNicks = Prelude.map nick $ Prelude.filter logonPassed $ elems clients |
1841
fba7210b438b
Retrieve client password from web database and ask for it
unc0rr
parents:
1839
diff
changeset
|
335 |
answerLobbyNicks = if not $ Prelude.null lobbyNicks then |
fba7210b438b
Retrieve client password from web database and ask for it
unc0rr
parents:
1839
diff
changeset
|
336 |
[AnswerThisClient (["LOBBY:JOINED"] ++ lobbyNicks)] |
fba7210b438b
Retrieve client password from web database and ask for it
unc0rr
parents:
1839
diff
changeset
|
337 |
else |
fba7210b438b
Retrieve client password from web database and ask for it
unc0rr
parents:
1839
diff
changeset
|
338 |
[] |
fba7210b438b
Retrieve client password from web database and ask for it
unc0rr
parents:
1839
diff
changeset
|
339 |
|
fba7210b438b
Retrieve client password from web database and ask for it
unc0rr
parents:
1839
diff
changeset
|
340 |
|
1862 | 341 |
processAction (clID, serverInfo, clients, rooms) (KickClient kickID) = do |
1879 | 342 |
liftM2 replaceID (return clID) (processAction (kickID, serverInfo, clients, rooms) $ ByeClient "Kicked") |
343 |
||
1926
cb46fbdcaa41
Add simple DoS protection mechanism (although better than previous server had)
unc0rr
parents:
1925
diff
changeset
|
344 |
|
1921 | 345 |
processAction (clID, serverInfo, clients, rooms) (BanClient banNick) = do |
346 |
return (clID, serverInfo, clients, rooms) |
|
347 |
||
1879 | 348 |
|
349 |
processAction (clID, serverInfo, clients, rooms) (KickRoomClient kickID) = do |
|
350 |
writeChan (sendChan $ clients ! kickID) ["KICKED"] |
|
2126 | 351 |
liftM2 replaceID (return clID) (processAction (kickID, serverInfo, clients, rooms) $ RoomRemoveThisClient "kicked") |
1879 | 352 |
|
353 |
||
354 |
processAction (clID, serverInfo, clients, rooms) (RemoveClientTeams teamsClID) = do |
|
355 |
liftM2 replaceID (return clID) $ |
|
356 |
foldM processAction (teamsClID, serverInfo, clients, rooms) $ removeTeamsActions |
|
357 |
where |
|
358 |
client = clients ! teamsClID |
|
359 |
room = rooms ! (roomID client) |
|
360 |
teamsToRemove = Prelude.filter (\t -> teamowner t == nick client) $ teams room |
|
361 |
removeTeamsActions = Prelude.map (RemoveTeam . teamname) teamsToRemove |
|
1926
cb46fbdcaa41
Add simple DoS protection mechanism (although better than previous server had)
unc0rr
parents:
1925
diff
changeset
|
362 |
|
cb46fbdcaa41
Add simple DoS protection mechanism (although better than previous server had)
unc0rr
parents:
1925
diff
changeset
|
363 |
|
cb46fbdcaa41
Add simple DoS protection mechanism (although better than previous server had)
unc0rr
parents:
1925
diff
changeset
|
364 |
processAction (clID, serverInfo, clients, rooms) (AddClient client) = do |
cb46fbdcaa41
Add simple DoS protection mechanism (although better than previous server had)
unc0rr
parents:
1925
diff
changeset
|
365 |
let updatedClients = insert (clientUID client) client clients |
1927
e2031906a347
Ping clients every 30 seconds. Disconnection due to ping timeout to be implemented.
unc0rr
parents:
1926
diff
changeset
|
366 |
infoM "Clients" ((show $ clientUID client) ++ ": New client. Time: " ++ (show $ connectTime client)) |
1926
cb46fbdcaa41
Add simple DoS protection mechanism (although better than previous server had)
unc0rr
parents:
1925
diff
changeset
|
367 |
writeChan (sendChan $ client) ["CONNECTED", "Hedgewars server http://www.hedgewars.org/"] |
cb46fbdcaa41
Add simple DoS protection mechanism (although better than previous server had)
unc0rr
parents:
1925
diff
changeset
|
368 |
|
1977 | 369 |
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
|
370 |
|
1931 | 371 |
if isJust $ host client `Prelude.lookup` newLogins then |
372 |
processAction (clID, serverInfo{lastLogins = newLogins}, updatedClients, rooms) $ ByeClient "Reconnected too fast" |
|
373 |
else |
|
374 |
return (clID, serverInfo{lastLogins = (host client, connectTime client) : newLogins}, updatedClients, rooms) |
|
1927
e2031906a347
Ping clients every 30 seconds. Disconnection due to ping timeout to be implemented.
unc0rr
parents:
1926
diff
changeset
|
375 |
|
e2031906a347
Ping clients every 30 seconds. Disconnection due to ping timeout to be implemented.
unc0rr
parents:
1926
diff
changeset
|
376 |
|
e2031906a347
Ping clients every 30 seconds. Disconnection due to ping timeout to be implemented.
unc0rr
parents:
1926
diff
changeset
|
377 |
processAction (clID, serverInfo, clients, rooms) PingAll = do |
1928 | 378 |
(_, _, newClients, newRooms) <- foldM kickTimeouted (clID, serverInfo, clients, rooms) $ elems clients |
1927
e2031906a347
Ping clients every 30 seconds. Disconnection due to ping timeout to be implemented.
unc0rr
parents:
1926
diff
changeset
|
379 |
processAction (clID, |
e2031906a347
Ping clients every 30 seconds. Disconnection due to ping timeout to be implemented.
unc0rr
parents:
1926
diff
changeset
|
380 |
serverInfo, |
1928 | 381 |
Data.IntMap.map (\cl -> cl{pingsQueue = pingsQueue cl + 1}) newClients, |
382 |
newRooms) $ AnswerAll ["PING"] |
|
383 |
where |
|
384 |
kickTimeouted (clID, serverInfo, clients, rooms) client = |
|
385 |
if pingsQueue client > 0 then |
|
386 |
processAction (clientUID client, serverInfo, clients, rooms) $ ByeClient "Ping timeout" |
|
387 |
else |
|
388 |
return (clID, serverInfo, clients, rooms) |
|
2172 | 389 |
|
390 |
||
391 |
processAction (clID, serverInfo, clients, rooms) (StatsAction) = do |
|
392 |
writeChan (dbQueries serverInfo) $ SendStats (size clients) (size rooms - 1) |
|
393 |
return (clID, serverInfo, clients, rooms) |