hedgewars: rust/hedgewars-server/src/server/handlers.rs@6a2e13e36b7f (annotated)

12149 44b06731278b Fix handlers module unc0rr parents: diff changeset	1	use mio;
13521 ba5211dddb21 Assorted chat fixes alfadur parents: 13444 diff changeset	2	use std::{io, io::Write};
12149 44b06731278b Fix handlers module unc0rr parents: diff changeset	3
14690 f61ce544d436 Server action refactoring part N of N alfadur <mail@none> parents: 14687 diff changeset	4	use super::{actions::Destination, core::HWServer, coretypes::ClientId};
14671 455865ccd36c Server action refactoring part 2 of N alfadur <mail@none> parents: 14457 diff changeset	5	use crate::{
14672 6e6632068a33 Server action refactoring part 3 of N alfadur <mail@none> parents: 14671 diff changeset	6	protocol::messages::{HWProtocolMessage, HWServerMessage, HWServerMessage::*},
14671 455865ccd36c Server action refactoring part 2 of N alfadur <mail@none> parents: 14457 diff changeset	7	server::actions::PendingMessage,
14693 6a2e13e36b7f add server anteroom alfadur <mail@none> parents: 14690 diff changeset	8	utils,
14671 455865ccd36c Server action refactoring part 2 of N alfadur <mail@none> parents: 14457 diff changeset	9	};
14693 6a2e13e36b7f add server anteroom alfadur <mail@none> parents: 14690 diff changeset	10	use base64::encode;
13805 0463a4221327 cleanup crate imports alfadur parents: 13798 diff changeset	11	use log::*;
14693 6a2e13e36b7f add server anteroom alfadur <mail@none> parents: 14690 diff changeset	12	use rand::{thread_rng, RngCore};
13666 09f4a30e50cc Rust 2018 conversion alfadur parents: 13521 diff changeset	13
14457 98ef2913ec73 Apply rustfmt to all files unc0rr parents: 14415 diff changeset	14	mod checker;
98ef2913ec73 Apply rustfmt to all files unc0rr parents: 14415 diff changeset	15	mod common;
12149 44b06731278b Fix handlers module unc0rr parents: diff changeset	16	mod inroom;
14457 98ef2913ec73 Apply rustfmt to all files unc0rr parents: 14415 diff changeset	17	mod lobby;
98ef2913ec73 Apply rustfmt to all files unc0rr parents: 14415 diff changeset	18	mod loggingin;
12149 44b06731278b Fix handlers module unc0rr parents: diff changeset	19
14693 6a2e13e36b7f add server anteroom alfadur <mail@none> parents: 14690 diff changeset	20	use self::loggingin::LoginResult;
6a2e13e36b7f add server anteroom alfadur <mail@none> parents: 14690 diff changeset	21
14671 455865ccd36c Server action refactoring part 2 of N alfadur <mail@none> parents: 14457 diff changeset	22	pub struct Response {
455865ccd36c Server action refactoring part 2 of N alfadur <mail@none> parents: 14457 diff changeset	23	client_id: ClientId,
455865ccd36c Server action refactoring part 2 of N alfadur <mail@none> parents: 14457 diff changeset	24	messages: Vec<PendingMessage>,
455865ccd36c Server action refactoring part 2 of N alfadur <mail@none> parents: 14457 diff changeset	25	}
455865ccd36c Server action refactoring part 2 of N alfadur <mail@none> parents: 14457 diff changeset	26
455865ccd36c Server action refactoring part 2 of N alfadur <mail@none> parents: 14457 diff changeset	27	impl Response {
455865ccd36c Server action refactoring part 2 of N alfadur <mail@none> parents: 14457 diff changeset	28	pub fn new(client_id: ClientId) -> Self {
455865ccd36c Server action refactoring part 2 of N alfadur <mail@none> parents: 14457 diff changeset	29	Self {
455865ccd36c Server action refactoring part 2 of N alfadur <mail@none> parents: 14457 diff changeset	30	client_id,
455865ccd36c Server action refactoring part 2 of N alfadur <mail@none> parents: 14457 diff changeset	31	messages: vec![],
455865ccd36c Server action refactoring part 2 of N alfadur <mail@none> parents: 14457 diff changeset	32	}
455865ccd36c Server action refactoring part 2 of N alfadur <mail@none> parents: 14457 diff changeset	33	}
455865ccd36c Server action refactoring part 2 of N alfadur <mail@none> parents: 14457 diff changeset	34
14672 6e6632068a33 Server action refactoring part 3 of N alfadur <mail@none> parents: 14671 diff changeset	35	#[inline]
6e6632068a33 Server action refactoring part 3 of N alfadur <mail@none> parents: 14671 diff changeset	36	pub fn is_empty(&self) -> bool {
6e6632068a33 Server action refactoring part 3 of N alfadur <mail@none> parents: 14671 diff changeset	37	self.messages.is_empty()
6e6632068a33 Server action refactoring part 3 of N alfadur <mail@none> parents: 14671 diff changeset	38	}
6e6632068a33 Server action refactoring part 3 of N alfadur <mail@none> parents: 14671 diff changeset	39
6e6632068a33 Server action refactoring part 3 of N alfadur <mail@none> parents: 14671 diff changeset	40	#[inline]
6e6632068a33 Server action refactoring part 3 of N alfadur <mail@none> parents: 14671 diff changeset	41	pub fn len(&self) -> usize {
6e6632068a33 Server action refactoring part 3 of N alfadur <mail@none> parents: 14671 diff changeset	42	self.messages.len()
6e6632068a33 Server action refactoring part 3 of N alfadur <mail@none> parents: 14671 diff changeset	43	}
6e6632068a33 Server action refactoring part 3 of N alfadur <mail@none> parents: 14671 diff changeset	44
6e6632068a33 Server action refactoring part 3 of N alfadur <mail@none> parents: 14671 diff changeset	45	#[inline]
14671 455865ccd36c Server action refactoring part 2 of N alfadur <mail@none> parents: 14457 diff changeset	46	pub fn client_id(&self) -> ClientId {
455865ccd36c Server action refactoring part 2 of N alfadur <mail@none> parents: 14457 diff changeset	47	self.client_id
455865ccd36c Server action refactoring part 2 of N alfadur <mail@none> parents: 14457 diff changeset	48	}
455865ccd36c Server action refactoring part 2 of N alfadur <mail@none> parents: 14457 diff changeset	49
14672 6e6632068a33 Server action refactoring part 3 of N alfadur <mail@none> parents: 14671 diff changeset	50	#[inline]
14671 455865ccd36c Server action refactoring part 2 of N alfadur <mail@none> parents: 14457 diff changeset	51	pub fn add(&mut self, message: PendingMessage) {
455865ccd36c Server action refactoring part 2 of N alfadur <mail@none> parents: 14457 diff changeset	52	self.messages.push(message)
455865ccd36c Server action refactoring part 2 of N alfadur <mail@none> parents: 14457 diff changeset	53	}
14672 6e6632068a33 Server action refactoring part 3 of N alfadur <mail@none> parents: 14671 diff changeset	54
6e6632068a33 Server action refactoring part 3 of N alfadur <mail@none> parents: 14671 diff changeset	55	pub fn extract_messages<'a, 'b: 'a>(
6e6632068a33 Server action refactoring part 3 of N alfadur <mail@none> parents: 14671 diff changeset	56	&'b mut self,
6e6632068a33 Server action refactoring part 3 of N alfadur <mail@none> parents: 14671 diff changeset	57	server: &'a HWServer,
6e6632068a33 Server action refactoring part 3 of N alfadur <mail@none> parents: 14671 diff changeset	58	) -> impl Iterator<Item = (Vec<ClientId>, HWServerMessage)> + 'a {
6e6632068a33 Server action refactoring part 3 of N alfadur <mail@none> parents: 14671 diff changeset	59	let client_id = self.client_id;
6e6632068a33 Server action refactoring part 3 of N alfadur <mail@none> parents: 14671 diff changeset	60	self.messages.drain(..).map(move \|m\| {
6e6632068a33 Server action refactoring part 3 of N alfadur <mail@none> parents: 14671 diff changeset	61	let ids = get_recipients(server, client_id, &m.destination);
6e6632068a33 Server action refactoring part 3 of N alfadur <mail@none> parents: 14671 diff changeset	62	(ids, m.message)
6e6632068a33 Server action refactoring part 3 of N alfadur <mail@none> parents: 14671 diff changeset	63	})
6e6632068a33 Server action refactoring part 3 of N alfadur <mail@none> parents: 14671 diff changeset	64	}
6e6632068a33 Server action refactoring part 3 of N alfadur <mail@none> parents: 14671 diff changeset	65	}
6e6632068a33 Server action refactoring part 3 of N alfadur <mail@none> parents: 14671 diff changeset	66
14687 5122c584804e Server action refactoring part A of N alfadur <mail@none> parents: 14673 diff changeset	67	impl Extend<PendingMessage> for Response {
5122c584804e Server action refactoring part A of N alfadur <mail@none> parents: 14673 diff changeset	68	fn extend<T: IntoIterator<Item = PendingMessage>>(&mut self, iter: T) {
5122c584804e Server action refactoring part A of N alfadur <mail@none> parents: 14673 diff changeset	69	for msg in iter {
5122c584804e Server action refactoring part A of N alfadur <mail@none> parents: 14673 diff changeset	70	self.add(msg)
5122c584804e Server action refactoring part A of N alfadur <mail@none> parents: 14673 diff changeset	71	}
5122c584804e Server action refactoring part A of N alfadur <mail@none> parents: 14673 diff changeset	72	}
5122c584804e Server action refactoring part A of N alfadur <mail@none> parents: 14673 diff changeset	73	}
5122c584804e Server action refactoring part A of N alfadur <mail@none> parents: 14673 diff changeset	74
14672 6e6632068a33 Server action refactoring part 3 of N alfadur <mail@none> parents: 14671 diff changeset	75	fn get_recipients(
6e6632068a33 Server action refactoring part 3 of N alfadur <mail@none> parents: 14671 diff changeset	76	server: &HWServer,
6e6632068a33 Server action refactoring part 3 of N alfadur <mail@none> parents: 14671 diff changeset	77	client_id: ClientId,
6e6632068a33 Server action refactoring part 3 of N alfadur <mail@none> parents: 14671 diff changeset	78	destination: &Destination,
6e6632068a33 Server action refactoring part 3 of N alfadur <mail@none> parents: 14671 diff changeset	79	) -> Vec<ClientId> {
6e6632068a33 Server action refactoring part 3 of N alfadur <mail@none> parents: 14671 diff changeset	80	let mut ids = match *destination {
6e6632068a33 Server action refactoring part 3 of N alfadur <mail@none> parents: 14671 diff changeset	81	Destination::ToSelf => vec![client_id],
6e6632068a33 Server action refactoring part 3 of N alfadur <mail@none> parents: 14671 diff changeset	82	Destination::ToId(id) => vec![id],
6e6632068a33 Server action refactoring part 3 of N alfadur <mail@none> parents: 14671 diff changeset	83	Destination::ToAll {
6e6632068a33 Server action refactoring part 3 of N alfadur <mail@none> parents: 14671 diff changeset	84	room_id: Some(id), ..
6e6632068a33 Server action refactoring part 3 of N alfadur <mail@none> parents: 14671 diff changeset	85	} => server.room_clients(id),
6e6632068a33 Server action refactoring part 3 of N alfadur <mail@none> parents: 14671 diff changeset	86	Destination::ToAll {
6e6632068a33 Server action refactoring part 3 of N alfadur <mail@none> parents: 14671 diff changeset	87	protocol: Some(proto),
6e6632068a33 Server action refactoring part 3 of N alfadur <mail@none> parents: 14671 diff changeset	88	..
6e6632068a33 Server action refactoring part 3 of N alfadur <mail@none> parents: 14671 diff changeset	89	} => server.protocol_clients(proto),
6e6632068a33 Server action refactoring part 3 of N alfadur <mail@none> parents: 14671 diff changeset	90	Destination::ToAll { .. } => server.clients.iter().map(\|(id, _)\| id).collect::<Vec<_>>(),
6e6632068a33 Server action refactoring part 3 of N alfadur <mail@none> parents: 14671 diff changeset	91	};
6e6632068a33 Server action refactoring part 3 of N alfadur <mail@none> parents: 14671 diff changeset	92	if let Destination::ToAll {
6e6632068a33 Server action refactoring part 3 of N alfadur <mail@none> parents: 14671 diff changeset	93	skip_self: true, ..
6e6632068a33 Server action refactoring part 3 of N alfadur <mail@none> parents: 14671 diff changeset	94	} = destination
6e6632068a33 Server action refactoring part 3 of N alfadur <mail@none> parents: 14671 diff changeset	95	{
6e6632068a33 Server action refactoring part 3 of N alfadur <mail@none> parents: 14671 diff changeset	96	if let Some(index) = ids.iter().position(\|id\| *id == client_id) {
6e6632068a33 Server action refactoring part 3 of N alfadur <mail@none> parents: 14671 diff changeset	97	ids.remove(index);
6e6632068a33 Server action refactoring part 3 of N alfadur <mail@none> parents: 14671 diff changeset	98	}
6e6632068a33 Server action refactoring part 3 of N alfadur <mail@none> parents: 14671 diff changeset	99	}
6e6632068a33 Server action refactoring part 3 of N alfadur <mail@none> parents: 14671 diff changeset	100	ids
14671 455865ccd36c Server action refactoring part 2 of N alfadur <mail@none> parents: 14457 diff changeset	101	}
455865ccd36c Server action refactoring part 2 of N alfadur <mail@none> parents: 14457 diff changeset	102
455865ccd36c Server action refactoring part 2 of N alfadur <mail@none> parents: 14457 diff changeset	103	pub fn handle(
455865ccd36c Server action refactoring part 2 of N alfadur <mail@none> parents: 14457 diff changeset	104	server: &mut HWServer,
455865ccd36c Server action refactoring part 2 of N alfadur <mail@none> parents: 14457 diff changeset	105	client_id: ClientId,
455865ccd36c Server action refactoring part 2 of N alfadur <mail@none> parents: 14457 diff changeset	106	response: &mut Response,
455865ccd36c Server action refactoring part 2 of N alfadur <mail@none> parents: 14457 diff changeset	107	message: HWProtocolMessage,
455865ccd36c Server action refactoring part 2 of N alfadur <mail@none> parents: 14457 diff changeset	108	) {
12149 44b06731278b Fix handlers module unc0rr parents: diff changeset	109	match message {
14693 6a2e13e36b7f add server anteroom alfadur <mail@none> parents: 14690 diff changeset	110	HWProtocolMessage::Ping => response.add(Pong.send_self()),
12149 44b06731278b Fix handlers module unc0rr parents: diff changeset	111	HWProtocolMessage::Malformed => warn!("Malformed/unknown message"),
44b06731278b Fix handlers module unc0rr parents: diff changeset	112	HWProtocolMessage::Empty => warn!("Empty message"),
14693 6a2e13e36b7f add server anteroom alfadur <mail@none> parents: 14690 diff changeset	113	_ => {
6a2e13e36b7f add server anteroom alfadur <mail@none> parents: 14690 diff changeset	114	if server.anteroom.clients.contains(client_id) {
6a2e13e36b7f add server anteroom alfadur <mail@none> parents: 14690 diff changeset	115	match loggingin::handle(&mut server.anteroom, client_id, response, message) {
6a2e13e36b7f add server anteroom alfadur <mail@none> parents: 14690 diff changeset	116	LoginResult::Unchanged => (),
6a2e13e36b7f add server anteroom alfadur <mail@none> parents: 14690 diff changeset	117	LoginResult::Complete => {
6a2e13e36b7f add server anteroom alfadur <mail@none> parents: 14690 diff changeset	118	if let Some(client) = server.anteroom.remove_client(client_id) {
6a2e13e36b7f add server anteroom alfadur <mail@none> parents: 14690 diff changeset	119	server.add_client(client_id, client);
6a2e13e36b7f add server anteroom alfadur <mail@none> parents: 14690 diff changeset	120	}
6a2e13e36b7f add server anteroom alfadur <mail@none> parents: 14690 diff changeset	121	}
6a2e13e36b7f add server anteroom alfadur <mail@none> parents: 14690 diff changeset	122	LoginResult::Exit => {
6a2e13e36b7f add server anteroom alfadur <mail@none> parents: 14690 diff changeset	123	server.anteroom.remove_client(client_id);
6a2e13e36b7f add server anteroom alfadur <mail@none> parents: 14690 diff changeset	124	}
6a2e13e36b7f add server anteroom alfadur <mail@none> parents: 14690 diff changeset	125	}
6a2e13e36b7f add server anteroom alfadur <mail@none> parents: 14690 diff changeset	126	} else {
6a2e13e36b7f add server anteroom alfadur <mail@none> parents: 14690 diff changeset	127	match message {
6a2e13e36b7f add server anteroom alfadur <mail@none> parents: 14690 diff changeset	128	HWProtocolMessage::Quit(Some(msg)) => {
6a2e13e36b7f add server anteroom alfadur <mail@none> parents: 14690 diff changeset	129	common::remove_client(server, response, "User quit: ".to_string() + &msg);
6a2e13e36b7f add server anteroom alfadur <mail@none> parents: 14690 diff changeset	130	}
6a2e13e36b7f add server anteroom alfadur <mail@none> parents: 14690 diff changeset	131	HWProtocolMessage::Quit(None) => {
6a2e13e36b7f add server anteroom alfadur <mail@none> parents: 14690 diff changeset	132	common::remove_client(server, response, "User quit".to_string());
6a2e13e36b7f add server anteroom alfadur <mail@none> parents: 14690 diff changeset	133	}
6a2e13e36b7f add server anteroom alfadur <mail@none> parents: 14690 diff changeset	134	_ => match server.clients[client_id].room_id {
6a2e13e36b7f add server anteroom alfadur <mail@none> parents: 14690 diff changeset	135	None => lobby::handle(server, client_id, response, message),
6a2e13e36b7f add server anteroom alfadur <mail@none> parents: 14690 diff changeset	136	Some(room_id) => {
6a2e13e36b7f add server anteroom alfadur <mail@none> parents: 14690 diff changeset	137	inroom::handle(server, client_id, response, room_id, message)
6a2e13e36b7f add server anteroom alfadur <mail@none> parents: 14690 diff changeset	138	}
6a2e13e36b7f add server anteroom alfadur <mail@none> parents: 14690 diff changeset	139	},
6a2e13e36b7f add server anteroom alfadur <mail@none> parents: 14690 diff changeset	140	}
14671 455865ccd36c Server action refactoring part 2 of N alfadur <mail@none> parents: 14457 diff changeset	141	}
14693 6a2e13e36b7f add server anteroom alfadur <mail@none> parents: 14690 diff changeset	142	}
12149 44b06731278b Fix handlers module unc0rr parents: diff changeset	143	}
44b06731278b Fix handlers module unc0rr parents: diff changeset	144	}
14673 08a8605bafaf Server action refactoring part 4 of N alfadur <mail@none> parents: 14672 diff changeset	145
14693 6a2e13e36b7f add server anteroom alfadur <mail@none> parents: 14690 diff changeset	146	pub fn handle_client_accept(server: &mut HWServer, client_id: ClientId, response: &mut Response) {
6a2e13e36b7f add server anteroom alfadur <mail@none> parents: 14690 diff changeset	147	let mut salt = [0u8; 18];
6a2e13e36b7f add server anteroom alfadur <mail@none> parents: 14690 diff changeset	148	thread_rng().fill_bytes(&mut salt);
6a2e13e36b7f add server anteroom alfadur <mail@none> parents: 14690 diff changeset	149
6a2e13e36b7f add server anteroom alfadur <mail@none> parents: 14690 diff changeset	150	server.anteroom.add_client(client_id, encode(&salt));
6a2e13e36b7f add server anteroom alfadur <mail@none> parents: 14690 diff changeset	151
6a2e13e36b7f add server anteroom alfadur <mail@none> parents: 14690 diff changeset	152	response.add(HWServerMessage::Connected(utils::PROTOCOL_VERSION).send_self());
6a2e13e36b7f add server anteroom alfadur <mail@none> parents: 14690 diff changeset	153	}
6a2e13e36b7f add server anteroom alfadur <mail@none> parents: 14690 diff changeset	154
14673 08a8605bafaf Server action refactoring part 4 of N alfadur <mail@none> parents: 14672 diff changeset	155	pub fn handle_client_loss(server: &mut HWServer, client_id: ClientId, response: &mut Response) {
08a8605bafaf Server action refactoring part 4 of N alfadur <mail@none> parents: 14672 diff changeset	156	common::remove_client(server, response, "Connection reset".to_string());
08a8605bafaf Server action refactoring part 4 of N alfadur <mail@none> parents: 14672 diff changeset	157	}

author	alfadur <mail@none>
	Thu, 07 Feb 2019 17:02:24 +0300
changeset 14693	6a2e13e36b7f
parent 14690	f61ce544d436
child 14694	25c564f77b7d
permissions	-rw-r--r--