author | unc0rr |
Tue, 19 Jun 2012 00:40:52 +0400 | |
changeset 7264 | 4c438ad3eddc |
parent 7126 | 8daa5c8e84c0 |
child 7321 | 57bd4f201401 |
permissions | -rw-r--r-- |
6012 | 1 |
{-# LANGUAGE CPP, OverloadedStrings #-} |
5184 | 2 |
module Actions where |
3 |
||
4 |
import Control.Concurrent |
|
5 |
import qualified Data.Set as Set |
|
6 |
import qualified Data.Sequence as Seq |
|
7 |
import qualified Data.List as L |
|
8 |
import qualified Control.Exception as Exception |
|
9 |
import System.Log.Logger |
|
10 |
import Control.Monad |
|
11 |
import Data.Time |
|
12 |
import Data.Maybe |
|
13 |
import Control.Monad.Reader |
|
14 |
import Control.Monad.State.Strict |
|
15 |
import qualified Data.ByteString.Char8 as B |
|
16 |
import Control.DeepSeq |
|
17 |
import Data.Unique |
|
18 |
import Control.Arrow |
|
19 |
import Control.Exception |
|
5209
f7a610e2ef5f
On restart command close server socket and spawn new server, keep running until last client quits
unc0rr
parents:
5184
diff
changeset
|
20 |
import System.Process |
f7a610e2ef5f
On restart command close server socket and spawn new server, keep running until last client quits
unc0rr
parents:
5184
diff
changeset
|
21 |
import Network.Socket |
5184 | 22 |
----------------------------- |
5996
2c72fe81dd37
Convert boolean variable + a bunch of fields which make sense only while game is going on into Maybe + structure
unc0rr
parents:
5426
diff
changeset
|
23 |
import OfficialServer.GameReplayStore |
5184 | 24 |
import CoreTypes |
25 |
import Utils |
|
26 |
import ClientIO |
|
27 |
import ServerState |
|
28 |
import Consts |
|
29 |
import ConfigFile |
|
6068 | 30 |
import EngineInteraction |
5184 | 31 |
|
32 |
data Action = |
|
33 |
AnswerClients ![ClientChan] ![B.ByteString] |
|
34 |
| SendServerMessage |
|
35 |
| SendServerVars |
|
36 |
| MoveToRoom RoomIndex |
|
37 |
| MoveToLobby B.ByteString |
|
38 |
| RemoveTeam B.ByteString |
|
6753
e95b1f62d0de
Don't remove client's teams from teams list on "ROUNDFINISHED 0", just send team removal message to others.
unc0rr
parents:
6733
diff
changeset
|
39 |
| SendTeamRemovalMessage B.ByteString |
5184 | 40 |
| RemoveRoom |
6758
26bf919aeb57
Oh, should also check for game finish when player quits without ROUNDFINISHED message: small refactoring, not tested at all
unc0rr
parents:
6756
diff
changeset
|
41 |
| FinishGame |
5184 | 42 |
| UnreadyRoomClients |
43 |
| JoinLobby |
|
44 |
| ProtocolError B.ByteString |
|
45 |
| Warning B.ByteString |
|
46 |
| NoticeMessage Notice |
|
47 |
| ByeClient B.ByteString |
|
48 |
| KickClient ClientIndex |
|
49 |
| KickRoomClient ClientIndex |
|
50 |
| BanClient NominalDiffTime B.ByteString ClientIndex |
|
5426
109e9b5761c2
Implement command for banning by ip and a command for bans list
unc0rr
parents:
5214
diff
changeset
|
51 |
| BanIP B.ByteString NominalDiffTime B.ByteString |
109e9b5761c2
Implement command for banning by ip and a command for bans list
unc0rr
parents:
5214
diff
changeset
|
52 |
| BanList |
5184 | 53 |
| ChangeMaster |
54 |
| RemoveClientTeams ClientIndex |
|
55 |
| ModifyClient (ClientInfo -> ClientInfo) |
|
56 |
| ModifyClient2 ClientIndex (ClientInfo -> ClientInfo) |
|
57 |
| ModifyRoom (RoomInfo -> RoomInfo) |
|
58 |
| ModifyServerInfo (ServerInfo -> ServerInfo) |
|
59 |
| AddRoom B.ByteString B.ByteString |
|
60 |
| CheckRegistered |
|
61 |
| ClearAccountsCache |
|
62 |
| ProcessAccountInfo AccountInfo |
|
63 |
| AddClient ClientInfo |
|
64 |
| DeleteClient ClientIndex |
|
65 |
| PingAll |
|
66 |
| StatsAction |
|
5209
f7a610e2ef5f
On restart command close server socket and spawn new server, keep running until last client quits
unc0rr
parents:
5184
diff
changeset
|
67 |
| RestartServer |
5184 | 68 |
| AddNick2Bans B.ByteString B.ByteString UTCTime |
69 |
| AddIP2Bans B.ByteString B.ByteString UTCTime |
|
70 |
| CheckBanned |
|
71 |
| SaveReplay |
|
72 |
||
73 |
||
74 |
type CmdHandler = [B.ByteString] -> Reader (ClientIndex, IRnC) [Action] |
|
75 |
||
76 |
instance NFData Action where |
|
77 |
rnf (AnswerClients chans msg) = chans `deepseq` msg `deepseq` () |
|
78 |
rnf a = a `seq` () |
|
79 |
||
80 |
instance NFData B.ByteString |
|
81 |
instance NFData (Chan a) |
|
82 |
||
83 |
||
84 |
othersChans :: StateT ServerState IO [ClientChan] |
|
85 |
othersChans = do |
|
86 |
cl <- client's id |
|
87 |
ri <- clientRoomA |
|
88 |
liftM (map sendChan . filter (/= cl)) $ roomClientsS ri |
|
89 |
||
90 |
processAction :: Action -> StateT ServerState IO () |
|
91 |
||
92 |
||
93 |
processAction (AnswerClients chans msg) = |
|
94 |
io $ mapM_ (`writeChan` (msg `deepseq` msg)) (chans `deepseq` chans) |
|
95 |
||
96 |
||
97 |
processAction SendServerMessage = do |
|
98 |
chan <- client's sendChan |
|
99 |
protonum <- client's clientProto |
|
100 |
si <- liftM serverInfo get |
|
101 |
let message = if protonum < latestReleaseVersion si then |
|
102 |
serverMessageForOldVersions si |
|
103 |
else |
|
104 |
serverMessage si |
|
105 |
processAction $ AnswerClients [chan] ["SERVER_MESSAGE", message] |
|
106 |
||
107 |
||
108 |
processAction SendServerVars = do |
|
109 |
chan <- client's sendChan |
|
110 |
si <- gets serverInfo |
|
111 |
io $ writeChan chan ("SERVER_VARS" : vars si) |
|
112 |
where |
|
113 |
vars si = [ |
|
114 |
"MOTD_NEW", serverMessage si, |
|
115 |
"MOTD_OLD", serverMessageForOldVersions si, |
|
116 |
"LATEST_PROTO", showB $ latestReleaseVersion si |
|
117 |
] |
|
118 |
||
119 |
||
120 |
processAction (ProtocolError msg) = do |
|
121 |
chan <- client's sendChan |
|
122 |
processAction $ AnswerClients [chan] ["ERROR", msg] |
|
123 |
||
124 |
||
125 |
processAction (Warning msg) = do |
|
126 |
chan <- client's sendChan |
|
127 |
processAction $ AnswerClients [chan] ["WARNING", msg] |
|
128 |
||
129 |
processAction (NoticeMessage n) = do |
|
130 |
chan <- client's sendChan |
|
131 |
processAction $ AnswerClients [chan] ["NOTICE", showB . fromEnum $ n] |
|
132 |
||
133 |
processAction (ByeClient msg) = do |
|
134 |
(Just ci) <- gets clientIndex |
|
135 |
ri <- clientRoomA |
|
136 |
||
137 |
chan <- client's sendChan |
|
138 |
clNick <- client's nick |
|
139 |
loggedIn <- client's logonPassed |
|
140 |
||
141 |
when (ri /= lobbyId) $ do |
|
142 |
processAction $ MoveToLobby ("quit: " `B.append` msg) |
|
143 |
return () |
|
144 |
||
145 |
clientsChans <- liftM (Prelude.map sendChan . Prelude.filter logonPassed) $! allClientsS |
|
146 |
io $ |
|
147 |
infoM "Clients" (show ci ++ " quits: " ++ B.unpack msg) |
|
148 |
||
149 |
processAction $ AnswerClients [chan] ["BYE", msg] |
|
150 |
when loggedIn $ processAction $ AnswerClients clientsChans ["LOBBY:LEFT", clNick, msg] |
|
151 |
||
152 |
s <- get |
|
153 |
put $! s{removedClients = ci `Set.insert` removedClients s} |
|
154 |
||
155 |
processAction (DeleteClient ci) = do |
|
156 |
io $ debugM "Clients" $ "DeleteClient: " ++ show ci |
|
157 |
||
158 |
rnc <- gets roomsClients |
|
159 |
io $ removeClient rnc ci |
|
160 |
||
161 |
s <- get |
|
162 |
put $! s{removedClients = ci `Set.delete` removedClients s} |
|
5209
f7a610e2ef5f
On restart command close server socket and spawn new server, keep running until last client quits
unc0rr
parents:
5184
diff
changeset
|
163 |
|
f7a610e2ef5f
On restart command close server socket and spawn new server, keep running until last client quits
unc0rr
parents:
5184
diff
changeset
|
164 |
sp <- gets (shutdownPending . serverInfo) |
f7a610e2ef5f
On restart command close server socket and spawn new server, keep running until last client quits
unc0rr
parents:
5184
diff
changeset
|
165 |
cls <- allClientsS |
f7a610e2ef5f
On restart command close server socket and spawn new server, keep running until last client quits
unc0rr
parents:
5184
diff
changeset
|
166 |
io $ when (sp && null cls) $ throwIO ShutdownException |
5184 | 167 |
|
168 |
processAction (ModifyClient f) = do |
|
169 |
(Just ci) <- gets clientIndex |
|
170 |
rnc <- gets roomsClients |
|
171 |
io $ modifyClient rnc f ci |
|
172 |
return () |
|
173 |
||
174 |
processAction (ModifyClient2 ci f) = do |
|
175 |
rnc <- gets roomsClients |
|
176 |
io $ modifyClient rnc f ci |
|
177 |
return () |
|
178 |
||
179 |
||
180 |
processAction (ModifyRoom f) = do |
|
181 |
rnc <- gets roomsClients |
|
182 |
ri <- clientRoomA |
|
183 |
io $ modifyRoom rnc f ri |
|
184 |
return () |
|
185 |
||
186 |
||
187 |
processAction (ModifyServerInfo f) = do |
|
188 |
modify (\s -> s{serverInfo = f $ serverInfo s}) |
|
189 |
si <- gets serverInfo |
|
190 |
io $ writeServerConfig si |
|
191 |
||
192 |
||
193 |
processAction (MoveToRoom ri) = do |
|
194 |
(Just ci) <- gets clientIndex |
|
195 |
rnc <- gets roomsClients |
|
196 |
||
197 |
io $ do |
|
198 |
modifyClient rnc (\cl -> cl{teamsInGame = 0, isReady = False, isMaster = False}) ci |
|
199 |
modifyRoom rnc (\r -> r{playersIn = playersIn r + 1}) ri |
|
200 |
moveClientToRoom rnc ri ci |
|
201 |
||
202 |
chans <- liftM (map sendChan) $ roomClientsS ri |
|
203 |
clNick <- client's nick |
|
204 |
||
205 |
processAction $ AnswerClients chans ["JOINED", clNick] |
|
206 |
||
207 |
||
208 |
processAction (MoveToLobby msg) = do |
|
209 |
(Just ci) <- gets clientIndex |
|
210 |
ri <- clientRoomA |
|
211 |
rnc <- gets roomsClients |
|
5996
2c72fe81dd37
Convert boolean variable + a bunch of fields which make sense only while game is going on into Maybe + structure
unc0rr
parents:
5426
diff
changeset
|
212 |
(gameProgress, playersNum) <- io $ room'sM rnc ((isJust . gameInfo) &&& playersIn) ri |
5184 | 213 |
ready <- client's isReady |
214 |
master <- client's isMaster |
|
215 |
-- client <- client's id |
|
216 |
clNick <- client's nick |
|
217 |
chans <- othersChans |
|
218 |
||
219 |
if master then |
|
220 |
if gameProgress && playersNum > 1 then |
|
221 |
mapM_ processAction [ChangeMaster, AnswerClients chans ["LEFT", clNick, msg], NoticeMessage AdminLeft, RemoveClientTeams ci] |
|
222 |
else |
|
223 |
processAction RemoveRoom |
|
224 |
else |
|
225 |
mapM_ processAction [AnswerClients chans ["LEFT", clNick, msg], RemoveClientTeams ci] |
|
226 |
||
227 |
-- when not removing room |
|
228 |
when (not master || (gameProgress && playersNum > 1)) . io $ do |
|
229 |
modifyRoom rnc (\r -> r{ |
|
230 |
playersIn = playersIn r - 1, |
|
231 |
readyPlayers = if ready then readyPlayers r - 1 else readyPlayers r |
|
232 |
}) ri |
|
233 |
moveClientToLobby rnc ci |
|
234 |
||
235 |
processAction ChangeMaster = do |
|
236 |
(Just ci) <- gets clientIndex |
|
237 |
ri <- clientRoomA |
|
238 |
rnc <- gets roomsClients |
|
239 |
newMasterId <- liftM (head . filter (/= ci)) . io $ roomClientsIndicesM rnc ri |
|
240 |
newMaster <- io $ client'sM rnc id newMasterId |
|
6733 | 241 |
oldRoomName <- io $ room'sM rnc name ri |
5184 | 242 |
let newRoomName = nick newMaster |
243 |
mapM_ processAction [ |
|
244 |
ModifyRoom (\r -> r{masterID = newMasterId, name = newRoomName}), |
|
245 |
ModifyClient2 newMasterId (\c -> c{isMaster = True}), |
|
246 |
AnswerClients [sendChan newMaster] ["ROOM_CONTROL_ACCESS", "1"] |
|
247 |
] |
|
248 |
||
6541
08ed346ed341
Send full room info on room add and update events. Less(?) traffic, but current frontend doesn't behave good with this change to server.
unc0rr
parents:
6191
diff
changeset
|
249 |
proto <- client's clientProto |
08ed346ed341
Send full room info on room add and update events. Less(?) traffic, but current frontend doesn't behave good with this change to server.
unc0rr
parents:
6191
diff
changeset
|
250 |
newRoom <- io $ room'sM rnc id ri |
08ed346ed341
Send full room info on room add and update events. Less(?) traffic, but current frontend doesn't behave good with this change to server.
unc0rr
parents:
6191
diff
changeset
|
251 |
chans <- liftM (map sendChan) $! sameProtoClientsS proto |
6733 | 252 |
processAction $ AnswerClients chans ("ROOM" : "UPD" : oldRoomName : roomInfo (nick newMaster) newRoom) |
6541
08ed346ed341
Send full room info on room add and update events. Less(?) traffic, but current frontend doesn't behave good with this change to server.
unc0rr
parents:
6191
diff
changeset
|
253 |
|
6758
26bf919aeb57
Oh, should also check for game finish when player quits without ROUNDFINISHED message: small refactoring, not tested at all
unc0rr
parents:
6756
diff
changeset
|
254 |
|
5184 | 255 |
processAction (AddRoom roomName roomPassword) = do |
256 |
Just clId <- gets clientIndex |
|
257 |
rnc <- gets roomsClients |
|
6541
08ed346ed341
Send full room info on room add and update events. Less(?) traffic, but current frontend doesn't behave good with this change to server.
unc0rr
parents:
6191
diff
changeset
|
258 |
proto <- client's clientProto |
08ed346ed341
Send full room info on room add and update events. Less(?) traffic, but current frontend doesn't behave good with this change to server.
unc0rr
parents:
6191
diff
changeset
|
259 |
n <- client's nick |
5184 | 260 |
|
261 |
let rm = newRoom{ |
|
262 |
masterID = clId, |
|
263 |
name = roomName, |
|
264 |
password = roomPassword, |
|
265 |
roomProto = proto |
|
266 |
} |
|
267 |
||
268 |
rId <- io $ addRoom rnc rm |
|
269 |
||
270 |
processAction $ MoveToRoom rId |
|
271 |
||
6541
08ed346ed341
Send full room info on room add and update events. Less(?) traffic, but current frontend doesn't behave good with this change to server.
unc0rr
parents:
6191
diff
changeset
|
272 |
chans <- liftM (map sendChan) $! sameProtoClientsS proto |
5184 | 273 |
|
274 |
mapM_ processAction [ |
|
6541
08ed346ed341
Send full room info on room add and update events. Less(?) traffic, but current frontend doesn't behave good with this change to server.
unc0rr
parents:
6191
diff
changeset
|
275 |
AnswerClients chans ("ROOM" : "ADD" : roomInfo n rm) |
5184 | 276 |
, ModifyClient (\cl -> cl{isMaster = True}) |
277 |
] |
|
278 |
||
279 |
||
280 |
processAction RemoveRoom = do |
|
281 |
Just clId <- gets clientIndex |
|
282 |
rnc <- gets roomsClients |
|
283 |
ri <- io $ clientRoomM rnc clId |
|
284 |
roomName <- io $ room'sM rnc name ri |
|
285 |
others <- othersChans |
|
6541
08ed346ed341
Send full room info on room add and update events. Less(?) traffic, but current frontend doesn't behave good with this change to server.
unc0rr
parents:
6191
diff
changeset
|
286 |
proto <- client's clientProto |
08ed346ed341
Send full room info on room add and update events. Less(?) traffic, but current frontend doesn't behave good with this change to server.
unc0rr
parents:
6191
diff
changeset
|
287 |
chans <- liftM (map sendChan) $! sameProtoClientsS proto |
5184 | 288 |
|
289 |
mapM_ processAction [ |
|
6541
08ed346ed341
Send full room info on room add and update events. Less(?) traffic, but current frontend doesn't behave good with this change to server.
unc0rr
parents:
6191
diff
changeset
|
290 |
AnswerClients chans ["ROOM", "DEL", roomName], |
5184 | 291 |
AnswerClients others ["ROOMABANDONED", roomName] |
292 |
] |
|
293 |
||
294 |
io $ removeRoom rnc ri |
|
295 |
||
296 |
||
6758
26bf919aeb57
Oh, should also check for game finish when player quits without ROUNDFINISHED message: small refactoring, not tested at all
unc0rr
parents:
6756
diff
changeset
|
297 |
processAction UnreadyRoomClients = do |
5184 | 298 |
rnc <- gets roomsClients |
299 |
ri <- clientRoomA |
|
300 |
roomPlayers <- roomClientsS ri |
|
301 |
roomClIDs <- io $ roomClientsIndicesM rnc ri |
|
302 |
pr <- client's clientProto |
|
303 |
processAction $ AnswerClients (map sendChan roomPlayers) $ notReadyMessage pr (map nick roomPlayers) |
|
304 |
io $ mapM_ (modifyClient rnc (\cl -> cl{isReady = False})) roomClIDs |
|
305 |
processAction $ ModifyRoom (\r -> r{readyPlayers = 0}) |
|
306 |
where |
|
307 |
notReadyMessage p nicks = if p < 38 then "NOT_READY" : nicks else "CLIENT_FLAGS" : "-r" : nicks |
|
308 |
||
6758
26bf919aeb57
Oh, should also check for game finish when player quits without ROUNDFINISHED message: small refactoring, not tested at all
unc0rr
parents:
6756
diff
changeset
|
309 |
|
26bf919aeb57
Oh, should also check for game finish when player quits without ROUNDFINISHED message: small refactoring, not tested at all
unc0rr
parents:
6756
diff
changeset
|
310 |
processAction FinishGame = do |
26bf919aeb57
Oh, should also check for game finish when player quits without ROUNDFINISHED message: small refactoring, not tested at all
unc0rr
parents:
6756
diff
changeset
|
311 |
rnc <- gets roomsClients |
26bf919aeb57
Oh, should also check for game finish when player quits without ROUNDFINISHED message: small refactoring, not tested at all
unc0rr
parents:
6756
diff
changeset
|
312 |
ri <- clientRoomA |
26bf919aeb57
Oh, should also check for game finish when player quits without ROUNDFINISHED message: small refactoring, not tested at all
unc0rr
parents:
6756
diff
changeset
|
313 |
thisRoomChans <- liftM (map sendChan) $ roomClientsS ri |
26bf919aeb57
Oh, should also check for game finish when player quits without ROUNDFINISHED message: small refactoring, not tested at all
unc0rr
parents:
6756
diff
changeset
|
314 |
clNick <- client's nick |
7126
8daa5c8e84c0
Bring leftTeams back (with a fix) as it is apparently needed for spectators.
unc0rr
parents:
7124
diff
changeset
|
315 |
answerRemovedTeams <- io $ |
8daa5c8e84c0
Bring leftTeams back (with a fix) as it is apparently needed for spectators.
unc0rr
parents:
7124
diff
changeset
|
316 |
room'sM rnc (map (\t -> AnswerClients thisRoomChans ["REMOVE_TEAM", t]) . leftTeams . fromJust . gameInfo) ri |
8daa5c8e84c0
Bring leftTeams back (with a fix) as it is apparently needed for spectators.
unc0rr
parents:
7124
diff
changeset
|
317 |
|
8daa5c8e84c0
Bring leftTeams back (with a fix) as it is apparently needed for spectators.
unc0rr
parents:
7124
diff
changeset
|
318 |
mapM_ processAction $ |
7124 | 319 |
SaveReplay |
7126
8daa5c8e84c0
Bring leftTeams back (with a fix) as it is apparently needed for spectators.
unc0rr
parents:
7124
diff
changeset
|
320 |
: ModifyRoom |
6758
26bf919aeb57
Oh, should also check for game finish when player quits without ROUNDFINISHED message: small refactoring, not tested at all
unc0rr
parents:
6756
diff
changeset
|
321 |
(\r -> r{ |
26bf919aeb57
Oh, should also check for game finish when player quits without ROUNDFINISHED message: small refactoring, not tested at all
unc0rr
parents:
6756
diff
changeset
|
322 |
gameInfo = Nothing, |
26bf919aeb57
Oh, should also check for game finish when player quits without ROUNDFINISHED message: small refactoring, not tested at all
unc0rr
parents:
6756
diff
changeset
|
323 |
readyPlayers = 0 |
26bf919aeb57
Oh, should also check for game finish when player quits without ROUNDFINISHED message: small refactoring, not tested at all
unc0rr
parents:
6756
diff
changeset
|
324 |
} |
26bf919aeb57
Oh, should also check for game finish when player quits without ROUNDFINISHED message: small refactoring, not tested at all
unc0rr
parents:
6756
diff
changeset
|
325 |
) |
7126
8daa5c8e84c0
Bring leftTeams back (with a fix) as it is apparently needed for spectators.
unc0rr
parents:
7124
diff
changeset
|
326 |
: UnreadyRoomClients |
8daa5c8e84c0
Bring leftTeams back (with a fix) as it is apparently needed for spectators.
unc0rr
parents:
7124
diff
changeset
|
327 |
: answerRemovedTeams |
5184 | 328 |
|
6758
26bf919aeb57
Oh, should also check for game finish when player quits without ROUNDFINISHED message: small refactoring, not tested at all
unc0rr
parents:
6756
diff
changeset
|
329 |
|
6753
e95b1f62d0de
Don't remove client's teams from teams list on "ROUNDFINISHED 0", just send team removal message to others.
unc0rr
parents:
6733
diff
changeset
|
330 |
processAction (SendTeamRemovalMessage teamName) = do |
e95b1f62d0de
Don't remove client's teams from teams list on "ROUNDFINISHED 0", just send team removal message to others.
unc0rr
parents:
6733
diff
changeset
|
331 |
chans <- othersChans |
e95b1f62d0de
Don't remove client's teams from teams list on "ROUNDFINISHED 0", just send team removal message to others.
unc0rr
parents:
6733
diff
changeset
|
332 |
mapM_ processAction [ |
e95b1f62d0de
Don't remove client's teams from teams list on "ROUNDFINISHED 0", just send team removal message to others.
unc0rr
parents:
6733
diff
changeset
|
333 |
AnswerClients chans ["EM", rmTeamMsg], |
e95b1f62d0de
Don't remove client's teams from teams list on "ROUNDFINISHED 0", just send team removal message to others.
unc0rr
parents:
6733
diff
changeset
|
334 |
ModifyRoom (\r -> r{ |
e95b1f62d0de
Don't remove client's teams from teams list on "ROUNDFINISHED 0", just send team removal message to others.
unc0rr
parents:
6733
diff
changeset
|
335 |
gameInfo = liftM (\g -> g{ |
7124 | 336 |
teamsInGameNumber = teamsInGameNumber g - 1 |
337 |
, roundMsgs = roundMsgs g Seq.|> rmTeamMsg |
|
6753
e95b1f62d0de
Don't remove client's teams from teams list on "ROUNDFINISHED 0", just send team removal message to others.
unc0rr
parents:
6733
diff
changeset
|
338 |
}) $ gameInfo r |
e95b1f62d0de
Don't remove client's teams from teams list on "ROUNDFINISHED 0", just send team removal message to others.
unc0rr
parents:
6733
diff
changeset
|
339 |
}) |
e95b1f62d0de
Don't remove client's teams from teams list on "ROUNDFINISHED 0", just send team removal message to others.
unc0rr
parents:
6733
diff
changeset
|
340 |
] |
6758
26bf919aeb57
Oh, should also check for game finish when player quits without ROUNDFINISHED message: small refactoring, not tested at all
unc0rr
parents:
6756
diff
changeset
|
341 |
|
26bf919aeb57
Oh, should also check for game finish when player quits without ROUNDFINISHED message: small refactoring, not tested at all
unc0rr
parents:
6756
diff
changeset
|
342 |
rnc <- gets roomsClients |
26bf919aeb57
Oh, should also check for game finish when player quits without ROUNDFINISHED message: small refactoring, not tested at all
unc0rr
parents:
6756
diff
changeset
|
343 |
ri <- clientRoomA |
26bf919aeb57
Oh, should also check for game finish when player quits without ROUNDFINISHED message: small refactoring, not tested at all
unc0rr
parents:
6756
diff
changeset
|
344 |
gi <- io $ room'sM rnc gameInfo ri |
26bf919aeb57
Oh, should also check for game finish when player quits without ROUNDFINISHED message: small refactoring, not tested at all
unc0rr
parents:
6756
diff
changeset
|
345 |
when (isJust gi && 0 == teamsInGameNumber (fromJust gi)) $ |
26bf919aeb57
Oh, should also check for game finish when player quits without ROUNDFINISHED message: small refactoring, not tested at all
unc0rr
parents:
6756
diff
changeset
|
346 |
processAction FinishGame |
6753
e95b1f62d0de
Don't remove client's teams from teams list on "ROUNDFINISHED 0", just send team removal message to others.
unc0rr
parents:
6733
diff
changeset
|
347 |
where |
e95b1f62d0de
Don't remove client's teams from teams list on "ROUNDFINISHED 0", just send team removal message to others.
unc0rr
parents:
6733
diff
changeset
|
348 |
rmTeamMsg = toEngineMsg $ 'F' `B.cons` teamName |
e95b1f62d0de
Don't remove client's teams from teams list on "ROUNDFINISHED 0", just send team removal message to others.
unc0rr
parents:
6733
diff
changeset
|
349 |
|
e95b1f62d0de
Don't remove client's teams from teams list on "ROUNDFINISHED 0", just send team removal message to others.
unc0rr
parents:
6733
diff
changeset
|
350 |
|
5184 | 351 |
processAction (RemoveTeam teamName) = do |
352 |
rnc <- gets roomsClients |
|
353 |
ri <- clientRoomA |
|
5996
2c72fe81dd37
Convert boolean variable + a bunch of fields which make sense only while game is going on into Maybe + structure
unc0rr
parents:
5426
diff
changeset
|
354 |
inGame <- io $ room'sM rnc (isJust . gameInfo) ri |
5184 | 355 |
chans <- othersChans |
7124 | 356 |
mapM_ processAction $ |
7126
8daa5c8e84c0
Bring leftTeams back (with a fix) as it is apparently needed for spectators.
unc0rr
parents:
7124
diff
changeset
|
357 |
ModifyRoom (\r -> r{ |
8daa5c8e84c0
Bring leftTeams back (with a fix) as it is apparently needed for spectators.
unc0rr
parents:
7124
diff
changeset
|
358 |
teams = Prelude.filter (\t -> teamName /= teamname t) $ teams r |
8daa5c8e84c0
Bring leftTeams back (with a fix) as it is apparently needed for spectators.
unc0rr
parents:
7124
diff
changeset
|
359 |
, gameInfo = liftM (\g -> g{leftTeams = teamName : leftTeams g}) $ gameInfo r |
8daa5c8e84c0
Bring leftTeams back (with a fix) as it is apparently needed for spectators.
unc0rr
parents:
7124
diff
changeset
|
360 |
}) |
7124 | 361 |
: AnswerClients chans ["REMOVE_TEAM", teamName] |
362 |
: [SendTeamRemovalMessage teamName | inGame] |
|
5184 | 363 |
|
364 |
||
365 |
processAction (RemoveClientTeams clId) = do |
|
366 |
rnc <- gets roomsClients |
|
367 |
||
368 |
removeTeamActions <- io $ do |
|
369 |
clNick <- client'sM rnc nick clId |
|
370 |
rId <- clientRoomM rnc clId |
|
371 |
roomTeams <- room'sM rnc teams rId |
|
372 |
return . Prelude.map (RemoveTeam . teamname) . Prelude.filter (\t -> teamowner t == clNick) $ roomTeams |
|
373 |
||
374 |
mapM_ processAction removeTeamActions |
|
375 |
||
376 |
||
377 |
||
378 |
processAction CheckRegistered = do |
|
379 |
(Just ci) <- gets clientIndex |
|
380 |
n <- client's nick |
|
381 |
h <- client's host |
|
382 |
p <- client's clientProto |
|
383 |
uid <- client's clUID |
|
6191 | 384 |
haveSameNick <- liftM (not . null . tail . filter (\c -> caseInsensitiveCompare (nick c) n)) allClientsS |
5184 | 385 |
if haveSameNick then |
386 |
if p < 38 then |
|
387 |
mapM_ processAction [ByeClient "Nickname is already in use", removeNick] |
|
388 |
else |
|
389 |
mapM_ processAction [NoticeMessage NickAlreadyInUse, removeNick] |
|
390 |
else |
|
391 |
do |
|
392 |
db <- gets (dbQueries . serverInfo) |
|
393 |
io $ writeChan db $ CheckAccount ci (hashUnique uid) n h |
|
394 |
return () |
|
395 |
where |
|
396 |
removeNick = ModifyClient (\c -> c{nick = ""}) |
|
397 |
||
398 |
||
399 |
processAction ClearAccountsCache = do |
|
400 |
dbq <- gets (dbQueries . serverInfo) |
|
401 |
io $ writeChan dbq ClearCache |
|
402 |
return () |
|
403 |
||
404 |
||
405 |
processAction (ProcessAccountInfo info) = |
|
406 |
case info of |
|
407 |
HasAccount passwd isAdmin -> do |
|
408 |
chan <- client's sendChan |
|
409 |
mapM_ processAction [AnswerClients [chan] ["ASKPASSWORD"], ModifyClient (\c -> c{webPassword = passwd, isAdministrator = isAdmin})] |
|
410 |
Guest -> |
|
411 |
processAction JoinLobby |
|
412 |
Admin -> do |
|
413 |
mapM_ processAction [ModifyClient (\cl -> cl{isAdministrator = True}), JoinLobby] |
|
414 |
chan <- client's sendChan |
|
415 |
processAction $ AnswerClients [chan] ["ADMIN_ACCESS"] |
|
416 |
||
417 |
||
418 |
processAction JoinLobby = do |
|
419 |
chan <- client's sendChan |
|
420 |
clientNick <- client's nick |
|
421 |
(lobbyNicks, clientsChans) <- liftM (unzip . Prelude.map (nick &&& sendChan) . Prelude.filter logonPassed) $! allClientsS |
|
422 |
mapM_ processAction $ |
|
423 |
AnswerClients clientsChans ["LOBBY:JOINED", clientNick] |
|
424 |
: AnswerClients [chan] ("LOBBY:JOINED" : clientNick : lobbyNicks) |
|
425 |
: [ModifyClient (\cl -> cl{logonPassed = True}), SendServerMessage] |
|
426 |
||
427 |
||
428 |
processAction (KickClient kickId) = do |
|
429 |
modify (\s -> s{clientIndex = Just kickId}) |
|
5214 | 430 |
clHost <- client's host |
431 |
currentTime <- io getCurrentTime |
|
432 |
mapM_ processAction [ |
|
433 |
AddIP2Bans clHost "60 seconds cooldown after kick" (addUTCTime 60 currentTime), |
|
434 |
ByeClient "Kicked" |
|
435 |
] |
|
5184 | 436 |
|
437 |
||
438 |
processAction (BanClient seconds reason banId) = do |
|
439 |
modify (\s -> s{clientIndex = Just banId}) |
|
440 |
clHost <- client's host |
|
441 |
currentTime <- io getCurrentTime |
|
5426
109e9b5761c2
Implement command for banning by ip and a command for bans list
unc0rr
parents:
5214
diff
changeset
|
442 |
let msg = B.concat ["Ban for ", B.pack . show $ seconds, " (", reason, ")"] |
5184 | 443 |
mapM_ processAction [ |
444 |
AddIP2Bans clHost msg (addUTCTime seconds currentTime) |
|
445 |
, KickClient banId |
|
446 |
] |
|
447 |
||
5426
109e9b5761c2
Implement command for banning by ip and a command for bans list
unc0rr
parents:
5214
diff
changeset
|
448 |
processAction (BanIP ip seconds reason) = do |
109e9b5761c2
Implement command for banning by ip and a command for bans list
unc0rr
parents:
5214
diff
changeset
|
449 |
currentTime <- io getCurrentTime |
109e9b5761c2
Implement command for banning by ip and a command for bans list
unc0rr
parents:
5214
diff
changeset
|
450 |
let msg = B.concat ["Ban for ", B.pack . show $ seconds, " (", reason, ")"] |
109e9b5761c2
Implement command for banning by ip and a command for bans list
unc0rr
parents:
5214
diff
changeset
|
451 |
processAction $ |
109e9b5761c2
Implement command for banning by ip and a command for bans list
unc0rr
parents:
5214
diff
changeset
|
452 |
AddIP2Bans ip msg (addUTCTime seconds currentTime) |
109e9b5761c2
Implement command for banning by ip and a command for bans list
unc0rr
parents:
5214
diff
changeset
|
453 |
|
109e9b5761c2
Implement command for banning by ip and a command for bans list
unc0rr
parents:
5214
diff
changeset
|
454 |
processAction BanList = do |
109e9b5761c2
Implement command for banning by ip and a command for bans list
unc0rr
parents:
5214
diff
changeset
|
455 |
ch <- client's sendChan |
109e9b5761c2
Implement command for banning by ip and a command for bans list
unc0rr
parents:
5214
diff
changeset
|
456 |
bans <- gets (bans . serverInfo) |
109e9b5761c2
Implement command for banning by ip and a command for bans list
unc0rr
parents:
5214
diff
changeset
|
457 |
processAction $ |
109e9b5761c2
Implement command for banning by ip and a command for bans list
unc0rr
parents:
5214
diff
changeset
|
458 |
AnswerClients [ch] ["BANLIST", B.pack $ show bans] |
109e9b5761c2
Implement command for banning by ip and a command for bans list
unc0rr
parents:
5214
diff
changeset
|
459 |
|
109e9b5761c2
Implement command for banning by ip and a command for bans list
unc0rr
parents:
5214
diff
changeset
|
460 |
|
5184 | 461 |
|
462 |
processAction (KickRoomClient kickId) = do |
|
463 |
modify (\s -> s{clientIndex = Just kickId}) |
|
464 |
ch <- client's sendChan |
|
465 |
mapM_ processAction [AnswerClients [ch] ["KICKED"], MoveToLobby "kicked"] |
|
466 |
||
467 |
||
468 |
processAction (AddClient cl) = do |
|
469 |
rnc <- gets roomsClients |
|
470 |
si <- gets serverInfo |
|
471 |
newClId <- io $ do |
|
472 |
ci <- addClient rnc cl |
|
473 |
_ <- Exception.mask (forkIO . clientRecvLoop (clientSocket cl) (coreChan si) (sendChan cl) ci) |
|
474 |
||
475 |
infoM "Clients" (show ci ++ ": New client. Time: " ++ show (connectTime cl)) |
|
476 |
||
477 |
return ci |
|
478 |
||
479 |
modify (\s -> s{clientIndex = Just newClId}) |
|
480 |
mapM_ processAction |
|
481 |
[ |
|
482 |
AnswerClients [sendChan cl] ["CONNECTED", "Hedgewars server http://www.hedgewars.org/", serverVersion] |
|
483 |
, CheckBanned |
|
6809 | 484 |
, AddIP2Bans (host cl) "Reconnected too fast" (addUTCTime 10 $ connectTime cl) |
5184 | 485 |
] |
486 |
||
487 |
||
488 |
processAction (AddNick2Bans n reason expiring) = do |
|
489 |
processAction $ ModifyServerInfo (\s -> s{bans = BanByNick n reason expiring : bans s}) |
|
490 |
||
491 |
processAction (AddIP2Bans ip reason expiring) = do |
|
492 |
(Just ci) <- gets clientIndex |
|
493 |
rc <- gets removedClients |
|
494 |
when (not $ ci `Set.member` rc) |
|
495 |
$ processAction $ ModifyServerInfo (\s -> s{bans = BanByIP ip reason expiring : bans s}) |
|
496 |
||
497 |
processAction CheckBanned = do |
|
498 |
clTime <- client's connectTime |
|
499 |
clNick <- client's nick |
|
500 |
clHost <- client's host |
|
501 |
si <- gets serverInfo |
|
502 |
let validBans = filter (checkNotExpired clTime) $ bans si |
|
503 |
let ban = L.find (checkBan clHost clNick) $ validBans |
|
5426
109e9b5761c2
Implement command for banning by ip and a command for bans list
unc0rr
parents:
5214
diff
changeset
|
504 |
mapM_ processAction $ |
5184 | 505 |
ModifyServerInfo (\s -> s{bans = validBans}) |
5426
109e9b5761c2
Implement command for banning by ip and a command for bans list
unc0rr
parents:
5214
diff
changeset
|
506 |
: [ByeClient (getBanReason $ fromJust ban) | isJust ban] |
5184 | 507 |
where |
508 |
checkNotExpired testTime (BanByIP _ _ time) = testTime `diffUTCTime` time <= 0 |
|
509 |
checkNotExpired testTime (BanByNick _ _ time) = testTime `diffUTCTime` time <= 0 |
|
5426
109e9b5761c2
Implement command for banning by ip and a command for bans list
unc0rr
parents:
5214
diff
changeset
|
510 |
checkBan ip _ (BanByIP bip _ _) = bip `B.isPrefixOf` ip |
5184 | 511 |
checkBan _ n (BanByNick bn _ _) = bn == n |
512 |
getBanReason (BanByIP _ msg _) = msg |
|
513 |
getBanReason (BanByNick _ msg _) = msg |
|
514 |
||
515 |
processAction PingAll = do |
|
516 |
rnc <- gets roomsClients |
|
517 |
io (allClientsM rnc) >>= mapM_ (kickTimeouted rnc) |
|
518 |
cis <- io $ allClientsM rnc |
|
519 |
chans <- io $ mapM (client'sM rnc sendChan) cis |
|
520 |
io $ mapM_ (modifyClient rnc (\cl -> cl{pingsQueue = pingsQueue cl + 1})) cis |
|
521 |
processAction $ AnswerClients chans ["PING"] |
|
522 |
where |
|
523 |
kickTimeouted rnc ci = do |
|
524 |
pq <- io $ client'sM rnc pingsQueue ci |
|
525 |
when (pq > 0) $ |
|
526 |
withStateT (\as -> as{clientIndex = Just ci}) $ |
|
527 |
processAction (ByeClient "Ping timeout") |
|
528 |
||
529 |
||
530 |
processAction StatsAction = do |
|
5211 | 531 |
si <- gets serverInfo |
532 |
when (not $ shutdownPending si) $ do |
|
5212
eaffb02f0053
Don't perform RestartServer action when already did it once
unc0rr
parents:
5211
diff
changeset
|
533 |
rnc <- gets roomsClients |
eaffb02f0053
Don't perform RestartServer action when already did it once
unc0rr
parents:
5211
diff
changeset
|
534 |
(roomsNum, clientsNum) <- io $ withRoomsAndClients rnc st |
eaffb02f0053
Don't perform RestartServer action when already did it once
unc0rr
parents:
5211
diff
changeset
|
535 |
io $ writeChan (dbQueries si) $ SendStats clientsNum (roomsNum - 1) |
5184 | 536 |
where |
537 |
st irnc = (length $ allRooms irnc, length $ allClients irnc) |
|
538 |
||
5212
eaffb02f0053
Don't perform RestartServer action when already did it once
unc0rr
parents:
5211
diff
changeset
|
539 |
processAction RestartServer = do |
eaffb02f0053
Don't perform RestartServer action when already did it once
unc0rr
parents:
5211
diff
changeset
|
540 |
sp <- gets (shutdownPending . serverInfo) |
eaffb02f0053
Don't perform RestartServer action when already did it once
unc0rr
parents:
5211
diff
changeset
|
541 |
when (not sp) $ do |
eaffb02f0053
Don't perform RestartServer action when already did it once
unc0rr
parents:
5211
diff
changeset
|
542 |
sock <- gets (fromJust . serverSocket . serverInfo) |
eaffb02f0053
Don't perform RestartServer action when already did it once
unc0rr
parents:
5211
diff
changeset
|
543 |
args <- gets (runArgs . serverInfo) |
eaffb02f0053
Don't perform RestartServer action when already did it once
unc0rr
parents:
5211
diff
changeset
|
544 |
io $ do |
eaffb02f0053
Don't perform RestartServer action when already did it once
unc0rr
parents:
5211
diff
changeset
|
545 |
noticeM "Core" "Closing listening socket" |
eaffb02f0053
Don't perform RestartServer action when already did it once
unc0rr
parents:
5211
diff
changeset
|
546 |
sClose sock |
eaffb02f0053
Don't perform RestartServer action when already did it once
unc0rr
parents:
5211
diff
changeset
|
547 |
noticeM "Core" "Spawning new server" |
eaffb02f0053
Don't perform RestartServer action when already did it once
unc0rr
parents:
5211
diff
changeset
|
548 |
_ <- createProcess (proc "./hedgewars-server" args) |
eaffb02f0053
Don't perform RestartServer action when already did it once
unc0rr
parents:
5211
diff
changeset
|
549 |
return () |
eaffb02f0053
Don't perform RestartServer action when already did it once
unc0rr
parents:
5211
diff
changeset
|
550 |
processAction $ ModifyServerInfo (\s -> s{shutdownPending = True}) |
5184 | 551 |
|
6012 | 552 |
#if defined(OFFICIAL_SERVER) |
5184 | 553 |
processAction SaveReplay = do |
554 |
ri <- clientRoomA |
|
555 |
rnc <- gets roomsClients |
|
556 |
io $ do |
|
557 |
r <- room'sM rnc id ri |
|
558 |
saveReplay r |
|
6012 | 559 |
#else |
560 |
processAction SaveReplay = return () |
|
561 |
#endif |