author | koda |
Fri, 02 Jul 2010 17:46:53 +0200 | |
changeset 3610 | 8590424bdfb0 |
parent 3568 | ae89cf0735dc |
child 3645 | c0b3f1bb9316 |
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 |
1804 | 10 |
import Monad |
1926
cb46fbdcaa41
Add simple DoS protection mechanism (although better than previous server had)
unc0rr
parents:
1925
diff
changeset
|
11 |
import Data.Time |
cb46fbdcaa41
Add simple DoS protection mechanism (although better than previous server had)
unc0rr
parents:
1925
diff
changeset
|
12 |
import Maybe |
3435 | 13 |
import Control.Monad.Reader |
3451
62089ccec75c
Uses StateT monad instead of manually maintaining the state
unc0rr
parents:
3436
diff
changeset
|
14 |
import Control.Monad.State |
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 |
||
3451
62089ccec75c
Uses StateT monad instead of manually maintaining the state
unc0rr
parents:
3436
diff
changeset
|
59 |
processAction (AnswerClients chans msg) = |
62089ccec75c
Uses StateT monad instead of manually maintaining the state
unc0rr
parents:
3436
diff
changeset
|
60 |
liftIO $ 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 |
3501 | 71 |
liftIO $ writeChan chan ["SERVER_MESSAGE", message] |
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 |
|
90 |
liftIO $ writeChan chan ["ERROR", msg] |
|
1804 | 91 |
|
92 |
||
3501 | 93 |
processAction (Warning msg) = do |
94 |
chan <- client's sendChan |
|
95 |
liftIO $ writeChan 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 |
3452 | 112 |
writeChan chan ["BYE", msg] |
3451
62089ccec75c
Uses StateT monad instead of manually maintaining the state
unc0rr
parents:
3436
diff
changeset
|
113 |
modifyRoom rnc (\r -> r{ |
62089ccec75c
Uses StateT monad instead of manually maintaining the state
unc0rr
parents:
3436
diff
changeset
|
114 |
--playersIDs = IntSet.delete ci (playersIDs r) |
3566 | 115 |
playersIn = (playersIn r) - 1, |
116 |
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
|
117 |
}) ri |
3566 | 118 |
|
119 |
removeClient rnc ci |
|
120 |
||
121 |
modify (\s -> s{removedClients = ci `Set.insert` removedClients s}) |
|
122 |
||
123 |
processAction (DeleteClient ci) = do |
|
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 |
184 |
||
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 |
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 |
} |
3502 | 254 |
|
255 |
rId <- liftIO $ addRoom rnc room |
|
256 |
||
257 |
chans <- liftM (map sendChan) $ roomClientsS lobbyId |
|
1804 | 258 |
|
3502 | 259 |
mapM_ processAction [ |
260 |
AnswerClients chans ["ROOM", "ADD", roomName] |
|
261 |
, ModifyClient (\cl -> cl{isMaster = True}) |
|
262 |
, MoveToRoom rId] |
|
1804 | 263 |
|
3502 | 264 |
{- |
3425 | 265 |
processAction (clID, serverInfo, rnc) (RemoveRoom) = do |
266 |
processAction (clID, serverInfo, rnc) $ AnswerLobby ["ROOM", "DEL", name room] |
|
267 |
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
|
268 |
return (clID, |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
269 |
serverInfo, |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
270 |
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
|
271 |
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
|
272 |
) |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
273 |
where |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
274 |
room = rooms ! rID |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
275 |
rID = roomID client |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
276 |
client = clients ! clID |
1804 | 277 |
|
1813 | 278 |
|
3425 | 279 |
processAction (clID, serverInfo, rnc) (UnreadyRoomClients) = do |
280 |
processAction (clID, serverInfo, rnc) $ AnswerThisRoom ("NOT_READY" : roomPlayers) |
|
2867
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
281 |
return (clID, |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
282 |
serverInfo, |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
283 |
Data.IntMap.map (\cl -> if roomID cl == rID then cl{isReady = False} else cl) clients, |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
284 |
adjust (\r -> r{readyPlayers = 0}) rID rooms) |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
285 |
where |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
286 |
room = rooms ! rID |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
287 |
rID = roomID client |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
288 |
client = clients ! clID |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
289 |
roomPlayers = Prelude.map (nick . (clients !)) roomPlayersIDs |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
290 |
roomPlayersIDs = IntSet.elems $ playersIDs room |
1811 | 291 |
|
292 |
||
3425 | 293 |
processAction (clID, serverInfo, rnc) (RemoveTeam teamName) = do |
2948
3f21a9dc93d0
Replace tabs with spaces using 'expand -t 4' command
unc0rr
parents:
2867
diff
changeset
|
294 |
newRooms <- if not $ gameinprogress room then |
2867
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
295 |
do |
3425 | 296 |
processAction (clID, serverInfo, rnc) $ AnswerOthersInRoom ["REMOVE_TEAM", teamName] |
2867
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
297 |
return $ |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
298 |
adjust (\r -> r{teams = Prelude.filter (\t -> teamName /= teamname t) $ teams r}) rID rooms |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
299 |
else |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
300 |
do |
3425 | 301 |
processAction (clID, serverInfo, rnc) $ AnswerOthersInRoom ["EM", rmTeamMsg] |
2867
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
302 |
return $ |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
303 |
adjust (\r -> r{ |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
304 |
teams = Prelude.filter (\t -> teamName /= teamname t) $ teams r, |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
305 |
leftTeams = teamName : leftTeams r, |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
306 |
roundMsgs = roundMsgs r Seq.|> rmTeamMsg |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
307 |
}) rID rooms |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
308 |
return (clID, serverInfo, clients, newRooms) |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
309 |
where |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
310 |
room = rooms ! rID |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
311 |
rID = roomID client |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
312 |
client = clients ! clID |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
313 |
rmTeamMsg = toEngineMsg $ 'F' : teamName |
3458 | 314 |
-} |
1813 | 315 |
|
3458 | 316 |
processAction CheckRegistered = do |
317 |
(Just ci) <- gets clientIndex |
|
3501 | 318 |
n <- client's nick |
319 |
h <- client's host |
|
3458 | 320 |
db <- gets (dbQueries . serverInfo) |
321 |
liftIO $ writeChan db $ CheckAccount ci n h |
|
322 |
return () |
|
1834 | 323 |
|
3458 | 324 |
{- |
3425 | 325 |
processAction (clID, serverInfo, rnc) (ClearAccountsCache) = do |
2867
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
326 |
writeChan (dbQueries serverInfo) ClearCache |
3425 | 327 |
return (clID, serverInfo, rnc) |
2867
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
328 |
where |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
329 |
client = clients ! clID |
2155
d897222d3339
Implement ability for server admin to clear accounts cache
unc0rr
parents:
2126
diff
changeset
|
330 |
|
d897222d3339
Implement ability for server admin to clear accounts cache
unc0rr
parents:
2126
diff
changeset
|
331 |
|
3425 | 332 |
processAction (clID, serverInfo, rnc) (Dump) = do |
2867
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
333 |
writeChan (sendChan $ clients ! clID) ["DUMP", show serverInfo, showTree clients, showTree rooms] |
3425 | 334 |
return (clID, serverInfo, rnc) |
3501 | 335 |
-} |
1841
fba7210b438b
Retrieve client password from web database and ask for it
unc0rr
parents:
1839
diff
changeset
|
336 |
|
3501 | 337 |
processAction (ProcessAccountInfo info) = |
2867
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
338 |
case info of |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
339 |
HasAccount passwd isAdmin -> do |
3501 | 340 |
chan <- client's sendChan |
341 |
liftIO $ writeChan chan ["ASKPASSWORD"] |
|
2867
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
342 |
Guest -> do |
3502 | 343 |
processAction JoinLobby |
2867
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
344 |
Admin -> do |
3502 | 345 |
mapM processAction [ModifyClient (\cl -> cl{isAdministrator = True}), JoinLobby] |
3501 | 346 |
chan <- client's sendChan |
347 |
liftIO $ writeChan chan ["ADMIN_ACCESS"] |
|
1841
fba7210b438b
Retrieve client password from web database and ask for it
unc0rr
parents:
1839
diff
changeset
|
348 |
|
3502 | 349 |
|
350 |
processAction JoinLobby = do |
|
3501 | 351 |
chan <- client's sendChan |
3502 | 352 |
clientNick <- client's nick |
3566 | 353 |
(lobbyNicks, clientsChans) <- liftM (unzip . Prelude.map (\c -> (nick c, sendChan c)) . Prelude.filter logonPassed) $! allClientsS |
3501 | 354 |
mapM_ processAction $ |
3502 | 355 |
(AnswerClients clientsChans ["LOBBY:JOINED", clientNick]) |
3566 | 356 |
: [AnswerClients [chan] ("LOBBY:JOINED" : clientNick : lobbyNicks)] |
3502 | 357 |
++ [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
|
358 |
|
3501 | 359 |
{- |
3502 | 360 |
processAction (clID, serverInfo, rnc) (RoomAddThisClient rID) = |
361 |
processAction ( |
|
362 |
clID, |
|
363 |
serverInfo, |
|
364 |
adjust (\cl -> cl{roomID = rID, teamsInGame = if rID == 0 then teamsInGame cl else 0}) clID clients, |
|
365 |
adjust (\r -> r{playersIDs = IntSet.insert clID (playersIDs r), playersIn = (playersIn r) + 1}) rID $ |
|
366 |
adjust (\r -> r{playersIDs = IntSet.delete clID (playersIDs r)}) 0 rooms |
|
367 |
) joinMsg |
|
368 |
where |
|
369 |
client = clients ! clID |
|
370 |
joinMsg = if rID == 0 then |
|
371 |
AnswerAllOthers ["LOBBY:JOINED", nick client] |
|
372 |
else |
|
373 |
AnswerThisRoom ["JOINED", nick client] |
|
1841
fba7210b438b
Retrieve client password from web database and ask for it
unc0rr
parents:
1839
diff
changeset
|
374 |
|
3425 | 375 |
processAction (clID, serverInfo, rnc) (KickClient kickID) = |
376 |
liftM2 replaceID (return clID) (processAction (kickID, serverInfo, rnc) $ ByeClient "Kicked") |
|
1879 | 377 |
|
1926
cb46fbdcaa41
Add simple DoS protection mechanism (although better than previous server had)
unc0rr
parents:
1925
diff
changeset
|
378 |
|
3425 | 379 |
processAction (clID, serverInfo, rnc) (BanClient banNick) = |
380 |
return (clID, serverInfo, rnc) |
|
1921 | 381 |
|
1879 | 382 |
|
3425 | 383 |
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
|
384 |
writeChan (sendChan $ clients ! kickID) ["KICKED"] |
3425 | 385 |
liftM2 replaceID (return clID) (processAction (kickID, serverInfo, rnc) $ RoomRemoveThisClient "kicked") |
1879 | 386 |
|
387 |
||
3425 | 388 |
processAction (clID, serverInfo, rnc) (RemoveClientTeams teamsClID) = |
2867
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
389 |
liftM2 replaceID (return clID) $ |
3425 | 390 |
foldM processAction (teamsClID, serverInfo, rnc) removeTeamsActions |
2867
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
391 |
where |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
392 |
client = clients ! teamsClID |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
393 |
room = rooms ! (roomID client) |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
394 |
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
|
395 |
removeTeamsActions = Prelude.map (RemoveTeam . teamname) teamsToRemove |
3435 | 396 |
-} |
1926
cb46fbdcaa41
Add simple DoS protection mechanism (although better than previous server had)
unc0rr
parents:
1925
diff
changeset
|
397 |
|
3451
62089ccec75c
Uses StateT monad instead of manually maintaining the state
unc0rr
parents:
3436
diff
changeset
|
398 |
processAction (AddClient client) = do |
62089ccec75c
Uses StateT monad instead of manually maintaining the state
unc0rr
parents:
3436
diff
changeset
|
399 |
rnc <- gets roomsClients |
62089ccec75c
Uses StateT monad instead of manually maintaining the state
unc0rr
parents:
3436
diff
changeset
|
400 |
si <- gets serverInfo |
62089ccec75c
Uses StateT monad instead of manually maintaining the state
unc0rr
parents:
3436
diff
changeset
|
401 |
liftIO $ do |
62089ccec75c
Uses StateT monad instead of manually maintaining the state
unc0rr
parents:
3436
diff
changeset
|
402 |
ci <- addClient rnc client |
3500
af8390d807d6
Use sockets instead of handles, use bytestrings instead of strings
unc0rr
parents:
3458
diff
changeset
|
403 |
forkIO $ clientRecvLoop (clientSocket client) (coreChan si) ci |
af8390d807d6
Use sockets instead of handles, use bytestrings instead of strings
unc0rr
parents:
3458
diff
changeset
|
404 |
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
|
405 |
|
62089ccec75c
Uses StateT monad instead of manually maintaining the state
unc0rr
parents:
3436
diff
changeset
|
406 |
infoM "Clients" (show ci ++ ": New client. Time: " ++ show (connectTime client)) |
62089ccec75c
Uses StateT monad instead of manually maintaining the state
unc0rr
parents:
3436
diff
changeset
|
407 |
writeChan (sendChan client) ["CONNECTED", "Hedgewars server http://www.hedgewars.org/"] |
3425 | 408 |
|
3451
62089ccec75c
Uses StateT monad instead of manually maintaining the state
unc0rr
parents:
3436
diff
changeset
|
409 |
{- 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
|
410 |
|
3451
62089ccec75c
Uses StateT monad instead of manually maintaining the state
unc0rr
parents:
3436
diff
changeset
|
411 |
if False && (isJust $ host client `Prelude.lookup` newLogins) then |
62089ccec75c
Uses StateT monad instead of manually maintaining the state
unc0rr
parents:
3436
diff
changeset
|
412 |
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
|
413 |
else |
62089ccec75c
Uses StateT monad instead of manually maintaining the state
unc0rr
parents:
3436
diff
changeset
|
414 |
return (ci, serverInfo) |
62089ccec75c
Uses StateT monad instead of manually maintaining the state
unc0rr
parents:
3436
diff
changeset
|
415 |
-} |
62089ccec75c
Uses StateT monad instead of manually maintaining the state
unc0rr
parents:
3436
diff
changeset
|
416 |
|
3458 | 417 |
|
3451
62089ccec75c
Uses StateT monad instead of manually maintaining the state
unc0rr
parents:
3436
diff
changeset
|
418 |
|
1927
e2031906a347
Ping clients every 30 seconds. Disconnection due to ping timeout to be implemented.
unc0rr
parents:
1926
diff
changeset
|
419 |
|
3435 | 420 |
{- |
3425 | 421 |
processAction (clID, serverInfo, rnc) PingAll = do |
422 |
(_, _, newClients, newRooms) <- foldM kickTimeouted (clID, serverInfo, rnc) $ elems clients |
|
2867
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
423 |
processAction (clID, |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
424 |
serverInfo, |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
425 |
Data.IntMap.map (\cl -> cl{pingsQueue = pingsQueue cl + 1}) newClients, |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
426 |
newRooms) $ AnswerAll ["PING"] |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
427 |
where |
3425 | 428 |
kickTimeouted (clID, serverInfo, rnc) client = |
2867
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
429 |
if pingsQueue client > 0 then |
3425 | 430 |
processAction (clientUID client, serverInfo, rnc) $ ByeClient "Ping timeout" |
2867
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
431 |
else |
3425 | 432 |
return (clID, serverInfo, rnc) |
2172 | 433 |
|
434 |
||
3425 | 435 |
processAction (clID, serverInfo, rnc) (StatsAction) = do |
2867
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2662
diff
changeset
|
436 |
writeChan (dbQueries serverInfo) $ SendStats (size clients) (size rooms - 1) |
3425 | 437 |
return (clID, serverInfo, rnc) |
438 |
||
3435 | 439 |
-} |