author | unc0rr |
Sun, 16 Dec 2018 00:12:29 +0100 | |
changeset 14478 | 98ef2913ec73 |
parent 14436 | 06672690d71b |
child 14525 | 6cc0fce249f9 |
permissions | -rw-r--r-- |
13416 | 1 |
use super::{ |
14478 | 2 |
actions, |
3 |
actions::{Destination, PendingMessage}, |
|
4 |
client::HWClient, |
|
5 |
coretypes::{ClientId, RoomId}, |
|
6 |
handlers, |
|
14413 | 7 |
io::HWServerIO, |
14478 | 8 |
room::HWRoom, |
13416 | 9 |
}; |
13666 | 10 |
use crate::protocol::messages::*; |
14478 | 11 |
use crate::utils; |
12 |
use base64::encode; |
|
13810 | 13 |
use log::*; |
14478 | 14 |
use rand::{thread_rng, RngCore}; |
15 |
use slab; |
|
12126 | 16 |
|
12852 | 17 |
type Slab<T> = slab::Slab<T>; |
12127 | 18 |
|
12126 | 19 |
pub struct HWServer { |
12143 | 20 |
pub clients: Slab<HWClient>, |
21 |
pub rooms: Slab<HWRoom>, |
|
13119
1e39b8749072
separated the server logic from all the async io mess.
alfadur
parents:
12852
diff
changeset
|
22 |
pub lobby_id: RoomId, |
13419 | 23 |
pub output: Vec<(Vec<ClientId>, HWServerMessage)>, |
13119
1e39b8749072
separated the server logic from all the async io mess.
alfadur
parents:
12852
diff
changeset
|
24 |
pub removed_clients: Vec<ClientId>, |
14478 | 25 |
pub io: Box<dyn HWServerIO>, |
12126 | 26 |
} |
27 |
||
28 |
impl HWServer { |
|
14413 | 29 |
pub fn new(clients_limit: usize, rooms_limit: usize, io: Box<dyn HWServerIO>) -> HWServer { |
13119
1e39b8749072
separated the server logic from all the async io mess.
alfadur
parents:
12852
diff
changeset
|
30 |
let rooms = Slab::with_capacity(rooms_limit); |
1e39b8749072
separated the server logic from all the async io mess.
alfadur
parents:
12852
diff
changeset
|
31 |
let clients = Slab::with_capacity(clients_limit); |
1e39b8749072
separated the server logic from all the async io mess.
alfadur
parents:
12852
diff
changeset
|
32 |
let mut server = HWServer { |
14478 | 33 |
clients, |
34 |
rooms, |
|
13119
1e39b8749072
separated the server logic from all the async io mess.
alfadur
parents:
12852
diff
changeset
|
35 |
lobby_id: 0, |
1e39b8749072
separated the server logic from all the async io mess.
alfadur
parents:
12852
diff
changeset
|
36 |
output: vec![], |
14413 | 37 |
removed_clients: vec![], |
14478 | 38 |
io, |
13119
1e39b8749072
separated the server logic from all the async io mess.
alfadur
parents:
12852
diff
changeset
|
39 |
}; |
1e39b8749072
separated the server logic from all the async io mess.
alfadur
parents:
12852
diff
changeset
|
40 |
server.lobby_id = server.add_room(); |
1e39b8749072
separated the server logic from all the async io mess.
alfadur
parents:
12852
diff
changeset
|
41 |
server |
12126 | 42 |
} |
43 |
||
13119
1e39b8749072
separated the server logic from all the async io mess.
alfadur
parents:
12852
diff
changeset
|
44 |
pub fn add_client(&mut self) -> ClientId { |
1e39b8749072
separated the server logic from all the async io mess.
alfadur
parents:
12852
diff
changeset
|
45 |
let key: ClientId; |
1e39b8749072
separated the server logic from all the async io mess.
alfadur
parents:
12852
diff
changeset
|
46 |
{ |
1e39b8749072
separated the server logic from all the async io mess.
alfadur
parents:
12852
diff
changeset
|
47 |
let entry = self.clients.vacant_entry(); |
1e39b8749072
separated the server logic from all the async io mess.
alfadur
parents:
12852
diff
changeset
|
48 |
key = entry.key(); |
13771 | 49 |
let mut salt = [0u8; 18]; |
50 |
thread_rng().fill_bytes(&mut salt); |
|
51 |
||
52 |
let client = HWClient::new(entry.key(), encode(&salt)); |
|
13119
1e39b8749072
separated the server logic from all the async io mess.
alfadur
parents:
12852
diff
changeset
|
53 |
entry.insert(client); |
1e39b8749072
separated the server logic from all the async io mess.
alfadur
parents:
12852
diff
changeset
|
54 |
} |
14478 | 55 |
self.send( |
56 |
key, |
|
57 |
&Destination::ToSelf, |
|
58 |
HWServerMessage::Connected(utils::PROTOCOL_VERSION), |
|
59 |
); |
|
13119
1e39b8749072
separated the server logic from all the async io mess.
alfadur
parents:
12852
diff
changeset
|
60 |
key |
12126 | 61 |
} |
12127 | 62 |
|
13119
1e39b8749072
separated the server logic from all the async io mess.
alfadur
parents:
12852
diff
changeset
|
63 |
pub fn client_lost(&mut self, client_id: ClientId) { |
14478 | 64 |
actions::run_action( |
65 |
self, |
|
66 |
client_id, |
|
67 |
actions::Action::ByeClient("Connection reset".to_string()), |
|
68 |
); |
|
12127 | 69 |
} |
70 |
||
13119
1e39b8749072
separated the server logic from all the async io mess.
alfadur
parents:
12852
diff
changeset
|
71 |
pub fn add_room(&mut self) -> RoomId { |
1e39b8749072
separated the server logic from all the async io mess.
alfadur
parents:
12852
diff
changeset
|
72 |
let entry = self.rooms.vacant_entry(); |
1e39b8749072
separated the server logic from all the async io mess.
alfadur
parents:
12852
diff
changeset
|
73 |
let key = entry.key(); |
1e39b8749072
separated the server logic from all the async io mess.
alfadur
parents:
12852
diff
changeset
|
74 |
let room = HWRoom::new(entry.key()); |
1e39b8749072
separated the server logic from all the async io mess.
alfadur
parents:
12852
diff
changeset
|
75 |
entry.insert(room); |
1e39b8749072
separated the server logic from all the async io mess.
alfadur
parents:
12852
diff
changeset
|
76 |
key |
12127 | 77 |
} |
12137
193dfdcb0620
- Use logging facilities instead of plain println!
unc0rr
parents:
12136
diff
changeset
|
78 |
|
13119
1e39b8749072
separated the server logic from all the async io mess.
alfadur
parents:
12852
diff
changeset
|
79 |
pub fn handle_msg(&mut self, client_id: ClientId, msg: HWProtocolMessage) { |
13416 | 80 |
debug!("Handling message {:?} for client {}", msg, client_id); |
13442
c6a3784ff2c1
Check for client's existence before handling messages
alfadur
parents:
13426
diff
changeset
|
81 |
if self.clients.contains(client_id) { |
c6a3784ff2c1
Check for client's existence before handling messages
alfadur
parents:
13426
diff
changeset
|
82 |
handlers::handle(self, client_id, msg); |
c6a3784ff2c1
Check for client's existence before handling messages
alfadur
parents:
13426
diff
changeset
|
83 |
} |
13119
1e39b8749072
separated the server logic from all the async io mess.
alfadur
parents:
12852
diff
changeset
|
84 |
} |
12139 | 85 |
|
13500 | 86 |
fn get_recipients(&self, client_id: ClientId, destination: &Destination) -> Vec<ClientId> { |
87 |
let mut ids = match *destination { |
|
13419 | 88 |
Destination::ToSelf => vec![client_id], |
13426 | 89 |
Destination::ToId(id) => vec![id], |
14478 | 90 |
Destination::ToAll { |
91 |
room_id: Some(id), .. |
|
92 |
} => self.room_clients(id), |
|
93 |
Destination::ToAll { |
|
94 |
protocol: Some(proto), |
|
95 |
.. |
|
96 |
} => self.protocol_clients(proto), |
|
97 |
Destination::ToAll { .. } => self.clients.iter().map(|(id, _)| id).collect::<Vec<_>>(), |
|
13419 | 98 |
}; |
14478 | 99 |
if let Destination::ToAll { |
100 |
skip_self: true, .. |
|
101 |
} = destination |
|
102 |
{ |
|
13419 | 103 |
if let Some(index) = ids.iter().position(|id| *id == client_id) { |
104 |
ids.remove(index); |
|
105 |
} |
|
106 |
} |
|
107 |
ids |
|
13416 | 108 |
} |
109 |
||
14478 | 110 |
pub fn send( |
111 |
&mut self, |
|
112 |
client_id: ClientId, |
|
113 |
destination: &Destination, |
|
114 |
message: HWServerMessage, |
|
115 |
) { |
|
13500 | 116 |
let ids = self.get_recipients(client_id, &destination); |
13419 | 117 |
self.output.push((ids, message)); |
13416 | 118 |
} |
119 |
||
13119
1e39b8749072
separated the server logic from all the async io mess.
alfadur
parents:
12852
diff
changeset
|
120 |
pub fn react(&mut self, client_id: ClientId, actions: Vec<actions::Action>) { |
12139 | 121 |
for action in actions { |
13119
1e39b8749072
separated the server logic from all the async io mess.
alfadur
parents:
12852
diff
changeset
|
122 |
actions::run_action(self, client_id, action); |
12138 | 123 |
} |
124 |
} |
|
13416 | 125 |
|
14478 | 126 |
pub fn lobby(&self) -> &HWRoom { |
127 |
&self.rooms[self.lobby_id] |
|
128 |
} |
|
13492 | 129 |
|
13416 | 130 |
pub fn has_room(&self, name: &str) -> bool { |
131 |
self.rooms.iter().any(|(_, r)| r.name == name) |
|
132 |
} |
|
133 |
||
134 |
pub fn find_room(&self, name: &str) -> Option<&HWRoom> { |
|
14478 | 135 |
self.rooms |
136 |
.iter() |
|
137 |
.find_map(|(_, r)| Some(r).filter(|r| r.name == name)) |
|
13416 | 138 |
} |
139 |
||
140 |
pub fn find_room_mut(&mut self, name: &str) -> Option<&mut HWRoom> { |
|
14478 | 141 |
self.rooms |
142 |
.iter_mut() |
|
143 |
.find_map(|(_, r)| Some(r).filter(|r| r.name == name)) |
|
13416 | 144 |
} |
145 |
||
13450 | 146 |
pub fn find_client(&self, nick: &str) -> Option<&HWClient> { |
14478 | 147 |
self.clients |
148 |
.iter() |
|
149 |
.find_map(|(_, c)| Some(c).filter(|c| c.nick == nick)) |
|
13450 | 150 |
} |
151 |
||
152 |
pub fn find_client_mut(&mut self, nick: &str) -> Option<&mut HWClient> { |
|
14478 | 153 |
self.clients |
154 |
.iter_mut() |
|
155 |
.find_map(|(_, c)| Some(c).filter(|c| c.nick == nick)) |
|
13450 | 156 |
} |
157 |
||
13416 | 158 |
pub fn select_clients<F>(&self, f: F) -> Vec<ClientId> |
14478 | 159 |
where |
160 |
F: Fn(&(usize, &HWClient)) -> bool, |
|
161 |
{ |
|
162 |
self.clients.iter().filter(f).map(|(_, c)| c.id).collect() |
|
13416 | 163 |
} |
164 |
||
165 |
pub fn room_clients(&self, room_id: RoomId) -> Vec<ClientId> { |
|
166 |
self.select_clients(|(_, c)| c.room_id == Some(room_id)) |
|
167 |
} |
|
168 |
||
13486 | 169 |
pub fn protocol_clients(&self, protocol: u16) -> Vec<ClientId> { |
13416 | 170 |
self.select_clients(|(_, c)| c.protocol_number == protocol) |
171 |
} |
|
172 |
||
173 |
pub fn other_clients_in_room(&self, self_id: ClientId) -> Vec<ClientId> { |
|
174 |
let room_id = self.clients[self_id].room_id; |
|
14478 | 175 |
self.select_clients(|(id, c)| *id != self_id && c.room_id == room_id) |
13416 | 176 |
} |
177 |
||
178 |
pub fn client_and_room(&mut self, client_id: ClientId) -> (&mut HWClient, Option<&mut HWRoom>) { |
|
179 |
let c = &mut self.clients[client_id]; |
|
180 |
if let Some(room_id) = c.room_id { |
|
181 |
(c, Some(&mut self.rooms[room_id])) |
|
182 |
} else { |
|
183 |
(c, None) |
|
184 |
} |
|
185 |
} |
|
13445
d3c86ade3d4d
Send the rnd reply to the room only.
Marcin Mielniczuk <marmistrz.dev@zoho.eu>
parents:
13442
diff
changeset
|
186 |
|
d3c86ade3d4d
Send the rnd reply to the room only.
Marcin Mielniczuk <marmistrz.dev@zoho.eu>
parents:
13442
diff
changeset
|
187 |
pub fn room(&mut self, client_id: ClientId) -> Option<&mut HWRoom> { |
d3c86ade3d4d
Send the rnd reply to the room only.
Marcin Mielniczuk <marmistrz.dev@zoho.eu>
parents:
13442
diff
changeset
|
188 |
self.client_and_room(client_id).1 |
d3c86ade3d4d
Send the rnd reply to the room only.
Marcin Mielniczuk <marmistrz.dev@zoho.eu>
parents:
13442
diff
changeset
|
189 |
} |
d3c86ade3d4d
Send the rnd reply to the room only.
Marcin Mielniczuk <marmistrz.dev@zoho.eu>
parents:
13442
diff
changeset
|
190 |
} |