diff -r 6843c4551cde -r 06672690d71b gameServer2/src/server/handlers/inroom.rs --- a/gameServer2/src/server/handlers/inroom.rs Mon Dec 10 21:23:52 2018 +0100 +++ /dev/null Thu Jan 01 00:00:00 1970 +0000 @@ -1,486 +0,0 @@ -use mio; - -use crate::{ - server::{ - coretypes::{ - ClientId, RoomId, Voting, VoteType, GameCfg, - MAX_HEDGEHOGS_PER_TEAM - }, - core::HWServer, - room::{HWRoom, RoomFlags}, - actions::{Action, Action::*} - }, - protocol::messages::{ - HWProtocolMessage, - HWServerMessage::*, - server_chat - }, - utils::is_name_illegal -}; -use std::{ - mem::swap -}; -use base64::{encode, decode}; -use super::common::rnd_reply; -use log::*; - -#[derive(Clone)] -struct ByMsg<'a> { - messages: &'a[u8] -} - -impl <'a> Iterator for ByMsg<'a> { - type Item = &'a[u8]; - - fn next(&mut self) -> Option<::Item> { - if let Some(size) = self.messages.get(0) { - let (msg, next) = self.messages.split_at(*size as usize + 1); - self.messages = next; - Some(msg) - } else { - None - } - } -} - -fn by_msg(source: &[u8]) -> ByMsg { - ByMsg {messages: source} -} - -const VALID_MESSAGES: &[u8] = - b"M#+LlRrUuDdZzAaSjJ,NpPwtgfhbc12345\x80\x81\x82\x83\x84\x85\x86\x87\x88\x89\x8A"; -const NON_TIMED_MESSAGES: &[u8] = b"M#hb"; - -#[cfg(canhazslicepatterns)] -fn is_msg_valid(msg: &[u8], team_indices: &[u8]) -> bool { - match msg { - [size, typ, body..] => VALID_MESSAGES.contains(typ) - && match body { - [1...MAX_HEDGEHOGS_PER_TEAM, team, ..] if *typ == b'h' => - team_indices.contains(team), - _ => *typ != b'h' - }, - _ => false - } -} - -fn is_msg_valid(msg: &[u8], _team_indices: &[u8]) -> bool { - if let Some(typ) = msg.get(1) { - VALID_MESSAGES.contains(typ) - } else { - false - } -} - -fn is_msg_empty(msg: &[u8]) -> bool { - msg.get(1).filter(|t| **t == b'+').is_some() -} - -fn is_msg_timed(msg: &[u8]) -> bool { - msg.get(1).filter(|t| !NON_TIMED_MESSAGES.contains(t)).is_some() -} - -fn voting_description(kind: &VoteType) -> String { - format!("New voting started: {}", match kind { - VoteType::Kick(nick) => format!("kick {}", nick), - VoteType::Map(name) => format!("map {}", name.as_ref().unwrap()), - VoteType::Pause => "pause".to_string(), - VoteType::NewSeed => "new seed".to_string(), - VoteType::HedgehogsPerTeam(number) => format!("hedgehogs per team: {}", number) - }) -} - -fn room_message_flag(msg: &HWProtocolMessage) -> RoomFlags { - use crate::protocol::messages::HWProtocolMessage::*; - match msg { - ToggleRestrictJoin => RoomFlags::RESTRICTED_JOIN, - ToggleRestrictTeams => RoomFlags::RESTRICTED_TEAM_ADD, - ToggleRegisteredOnly => RoomFlags::RESTRICTED_UNREGISTERED_PLAYERS, - _ => RoomFlags::empty() - } -} - -pub fn handle(server: &mut HWServer, client_id: ClientId, room_id: RoomId, message: HWProtocolMessage) { - use crate::protocol::messages::HWProtocolMessage::*; - match message { - Part(None) => server.react(client_id, vec![ - MoveToLobby("part".to_string())]), - Part(Some(msg)) => server.react(client_id, vec![ - MoveToLobby(format!("part: {}", msg))]), - Chat(msg) => { - let actions = { - let c = &mut server.clients[client_id]; - let chat_msg = ChatMsg {nick: c.nick.clone(), msg}; - vec![chat_msg.send_all().in_room(room_id).but_self().action()] - }; - server.react(client_id, actions); - }, - Fix => { - if let (c, Some(r)) = server.client_and_room(client_id) { - if c.is_admin() { r.set_is_fixed(true) } - } - } - Unfix => { - if let (c, Some(r)) = server.client_and_room(client_id) { - if c.is_admin() { r.set_is_fixed(false) } - } - } - Greeting(text) => { - if let (c, Some(r)) = server.client_and_room(client_id) { - if c.is_admin() || c.is_master() && !r.is_fixed() { - r.greeting = text - } - } - } - RoomName(new_name) => { - let actions = - if is_name_illegal(&new_name) { - vec![Warn("Illegal room name! A room name must be between 1-40 characters long, must not have a trailing or leading space and must not have any of these characters: $()*+?[]^{|}".to_string())] - } else if server.rooms[room_id].is_fixed() { - vec![Warn("Access denied.".to_string())] - } else if server.has_room(&new_name) { - vec![Warn("A room with the same name already exists.".to_string())] - } else { - let mut old_name = new_name.clone(); - swap(&mut server.rooms[room_id].name, &mut old_name); - vec![SendRoomUpdate(Some(old_name))] - }; - server.react(client_id, actions); - }, - ToggleReady => { - if let (c, Some(r)) = server.client_and_room(client_id) { - let flags = if c.is_ready() { - r.ready_players_number -= 1; - "-r" - } else { - r.ready_players_number += 1; - "+r" - }; - - let msg = if c.protocol_number < 38 { - LegacyReady(c.is_ready(), vec![c.nick.clone()]) - } else { - ClientFlags(flags.to_string(), vec![c.nick.clone()]) - }; - - let mut v = vec![msg.send_all().in_room(r.id).action()]; - - if r.is_fixed() && r.ready_players_number == r.players_number { - v.push(StartRoomGame(r.id)) - } - - c.set_is_ready(!c.is_ready()); - server.react(client_id, v); - } - } - AddTeam(info) => { - let mut actions = Vec::new(); - if let (c, Some(r)) = server.client_and_room(client_id) { - if r.teams.len() >= r.team_limit as usize { - actions.push(Warn("Too many teams!".to_string())) - } else if r.addable_hedgehogs() == 0 { - actions.push(Warn("Too many hedgehogs!".to_string())) - } else if r.find_team(|t| t.name == info.name) != None { - actions.push(Warn("There's already a team with same name in the list.".to_string())) - } else if r.game_info.is_some() { - actions.push(Warn("Joining not possible: Round is in progress.".to_string())) - } else if r.is_team_add_restricted() { - actions.push(Warn("This room currently does not allow adding new teams.".to_string())); - } else { - let team = r.add_team(c.id, *info, c.protocol_number < 42); - c.teams_in_game += 1; - c.clan = Some(team.color); - actions.push(TeamAccepted(team.name.clone()) - .send_self().action()); - actions.push(TeamAdd(HWRoom::team_info(&c, team)) - .send_all().in_room(room_id).but_self().action()); - actions.push(TeamColor(team.name.clone(), team.color) - .send_all().in_room(room_id).action()); - actions.push(HedgehogsNumber(team.name.clone(), team.hedgehogs_number) - .send_all().in_room(room_id).action()); - actions.push(SendRoomUpdate(None)); - } - } - server.react(client_id, actions); - }, - RemoveTeam(name) => { - let mut actions = Vec::new(); - if let (c, Some(r)) = server.client_and_room(client_id) { - match r.find_team_owner(&name) { - None => - actions.push(Warn("Error: The team you tried to remove does not exist.".to_string())), - Some((id, _)) if id != client_id => - actions.push(Warn("You can't remove a team you don't own.".to_string())), - Some((_, name)) => { - c.teams_in_game -= 1; - c.clan = r.find_team_color(c.id); - actions.push(Action::RemoveTeam(name.to_string())); - } - } - }; - server.react(client_id, actions); - }, - SetHedgehogsNumber(team_name, number) => { - if let (c, Some(r)) = server.client_and_room(client_id) { - let addable_hedgehogs = r.addable_hedgehogs(); - let actions = if let Some((_, team)) = r.find_team_and_owner_mut(|t| t.name == team_name) { - if !c.is_master() { - vec![ProtocolError("You're not the room master!".to_string())] - } else if number < 1 || number > MAX_HEDGEHOGS_PER_TEAM - || number > addable_hedgehogs + team.hedgehogs_number { - vec![HedgehogsNumber(team.name.clone(), team.hedgehogs_number) - .send_self().action()] - } else { - team.hedgehogs_number = number; - vec![HedgehogsNumber(team.name.clone(), number) - .send_all().in_room(room_id).but_self().action()] - } - } else { - vec![(Warn("No such team.".to_string()))] - }; - server.react(client_id, actions); - } - }, - SetTeamColor(team_name, color) => { - if let (c, Some(r)) = server.client_and_room(client_id) { - let mut owner_id = None; - let actions = if let Some((owner, team)) = r.find_team_and_owner_mut(|t| t.name == team_name) { - if !c.is_master() { - vec![ProtocolError("You're not the room master!".to_string())] - } else if false { - Vec::new() - } else { - owner_id = Some(owner); - team.color = color; - vec![TeamColor(team.name.clone(), color) - .send_all().in_room(room_id).but_self().action()] - } - } else { - vec![(Warn("No such team.".to_string()))] - }; - - if let Some(id) = owner_id { - server.clients[id].clan = Some(color); - } - - server.react(client_id, actions); - }; - }, - Cfg(cfg) => { - if let (c, Some(r)) = server.client_and_room(client_id) { - let actions = if r.is_fixed() { - vec![Warn("Access denied.".to_string())] - } else if !c.is_master() { - vec![ProtocolError("You're not the room master!".to_string())] - } else { - let cfg = match cfg { - GameCfg::Scheme(name, mut values) => { - if c.protocol_number == 49 && values.len() >= 2 { - let mut s = "X".repeat(50); - s.push_str(&values.pop().unwrap()); - values.push(s); - } - GameCfg::Scheme(name, values) - } - cfg => cfg - }; - - let v = vec![cfg.to_server_msg() - .send_all().in_room(r.id).but_self().action()]; - r.set_config(cfg); - v - }; - server.react(client_id, actions); - } - } - Save(name, location) => { - let actions = vec![server_chat(format!("Room config saved as {}", name)) - .send_all().in_room(room_id).action()]; - server.rooms[room_id].save_config(name, location); - server.react(client_id, actions); - } - SaveRoom(filename) => { - if server.clients[client_id].is_admin() { - let actions = match server.rooms[room_id].get_saves() { - Ok(text) => match server.io.write_file(&filename, &text) { - Ok(_) => vec![server_chat("Room configs saved successfully.".to_string()) - .send_self().action()], - Err(e) => { - warn!("Error while writing the config file \"{}\": {}", filename, e); - vec![Warn("Unable to save the room configs.".to_string())] - } - } - Err(e) => { - warn!("Error while serializing the room configs: {}", e); - vec![Warn("Unable to serialize the room configs.".to_string())] - } - }; - server.react(client_id, actions); - } - } - LoadRoom(filename) => { - if server.clients[client_id].is_admin() { - let actions = match server.io.read_file(&filename) { - Ok(text) => match server.rooms[room_id].set_saves(&text) { - Ok(_) => vec![server_chat("Room configs loaded successfully.".to_string()) - .send_self().action()], - Err(e) => { - warn!("Error while deserializing the room configs: {}", e); - vec![Warn("Unable to deserialize the room configs.".to_string())] - } - } - Err(e) => { - warn!("Error while reading the config file \"{}\": {}", filename, e); - vec![Warn("Unable to load the room configs.".to_string())] - } - }; - server.react(client_id, actions); - } - } - Delete(name) => { - let actions = if !server.rooms[room_id].delete_config(&name) { - vec![Warn(format!("Save doesn't exist: {}", name))] - } else { - vec![server_chat(format!("Room config {} has been deleted", name)) - .send_all().in_room(room_id).action()] - }; - server.react(client_id, actions); - } - CallVote(None) => { - server.react(client_id, vec![ - server_chat("Available callvote commands: kick , map , pause, newseed, hedgehogs ".to_string()) - .send_self().action()]) - } - CallVote(Some(kind)) => { - let is_in_game = server.rooms[room_id].game_info.is_some(); - let error = match &kind { - VoteType::Kick(nick) => { - if server.find_client(&nick).filter(|c| c.room_id == Some(room_id)).is_some() { - None - } else { - Some("/callvote kick: No such user!".to_string()) - } - }, - VoteType::Map(None) => { - let names: Vec<_> = server.rooms[room_id].saves.keys().cloned().collect(); - if names.is_empty() { - Some("/callvote map: No maps saved in this room!".to_string()) - } else { - Some(format!("Available maps: {}", names.join(", "))) - } - }, - VoteType::Map(Some(name)) => { - if server.rooms[room_id].saves.get(&name[..]).is_some() { - None - } else { - Some("/callvote map: No such map!".to_string()) - } - }, - VoteType::Pause => { - if is_in_game { - None - } else { - Some("/callvote pause: No game in progress!".to_string()) - } - }, - VoteType::NewSeed => { - None - }, - VoteType::HedgehogsPerTeam(number) => { - match number { - 1...MAX_HEDGEHOGS_PER_TEAM => None, - _ => Some("/callvote hedgehogs: Specify number from 1 to 8.".to_string()) - } - }, - }; - match error { - None => { - let msg = voting_description(&kind); - let voting = Voting::new(kind, server.room_clients(client_id)); - server.rooms[room_id].voting = Some(voting); - server.react(client_id, vec![ - server_chat(msg).send_all().in_room(room_id).action(), - AddVote{ vote: true, is_forced: false}]); - } - Some(msg) => { - server.react(client_id, vec![ - server_chat(msg).send_self().action()]) - } - } - } - Vote(vote) => { - server.react(client_id, vec![AddVote{ vote, is_forced: false }]); - } - ForceVote(vote) => { - let is_forced = server.clients[client_id].is_admin(); - server.react(client_id, vec![AddVote{ vote, is_forced }]); - } - ToggleRestrictJoin | ToggleRestrictTeams | ToggleRegisteredOnly => { - if server.clients[client_id].is_master() { - server.rooms[room_id].flags.toggle(room_message_flag(&message)); - } - server.react(client_id, vec![SendRoomUpdate(None)]); - } - StartGame => { - server.react(client_id, vec![StartRoomGame(room_id)]); - } - EngineMessage(em) => { - let mut actions = Vec::new(); - if let (c, Some(r)) = server.client_and_room(client_id) { - if c.teams_in_game > 0 { - let decoding = decode(&em[..]).unwrap(); - let messages = by_msg(&decoding); - let valid = messages.filter(|m| is_msg_valid(m, &c.team_indices)); - let non_empty = valid.clone().filter(|m| !is_msg_empty(m)); - let sync_msg = valid.clone().filter(|m| is_msg_timed(m)) - .last().map(|m| if is_msg_empty(m) {Some(encode(m))} else {None}); - - let em_response = encode(&valid.flat_map(|msg| msg).cloned().collect::>()); - if !em_response.is_empty() { - actions.push(ForwardEngineMessage(vec![em_response]) - .send_all().in_room(r.id).but_self().action()); - } - let em_log = encode(&non_empty.flat_map(|msg| msg).cloned().collect::>()); - if let Some(ref mut info) = r.game_info { - if !em_log.is_empty() { - info.msg_log.push(em_log); - } - if let Some(msg) = sync_msg { - info.sync_msg = msg; - } - } - } - } - server.react(client_id, actions) - } - RoundFinished => { - let mut actions = Vec::new(); - if let (c, Some(r)) = server.client_and_room(client_id) { - if c.is_in_game() { - c.set_is_in_game(false); - actions.push(ClientFlags("-g".to_string(), vec![c.nick.clone()]). - send_all().in_room(r.id).action()); - if r.game_info.is_some() { - for team in r.client_teams(c.id) { - actions.push(SendTeamRemovalMessage(team.name.clone())); - } - } - } - } - server.react(client_id, actions) - }, - Rnd(v) => { - let result = rnd_reply(&v); - let mut echo = vec!["/rnd".to_string()]; - echo.extend(v.into_iter()); - let chat_msg = ChatMsg { - nick: server.clients[client_id].nick.clone(), - msg: echo.join(" ") - }; - server.react(client_id, vec![ - chat_msg.send_all().in_room(room_id).action(), - result.send_all().in_room(room_id).action()]) - }, - _ => warn!("Unimplemented!") - } -}