disallow mutable rooms outside the server
authoralfadur <mail@none>
Tue, 24 Dec 2019 21:45:26 +0300
changeset 15526 24f692e791d3
parent 15525 16d3c9acd715
child 15527 428a0e7da27b
disallow mutable rooms outside the server
rust/hedgewars-server/src/core/server.rs
rust/hedgewars-server/src/handlers.rs
rust/hedgewars-server/src/handlers/common.rs
rust/hedgewars-server/src/handlers/inlobby.rs
--- a/rust/hedgewars-server/src/core/server.rs	Tue Dec 24 20:57:58 2019 +0300
+++ b/rust/hedgewars-server/src/core/server.rs	Tue Dec 24 21:45:26 2019 +0300
@@ -161,9 +161,9 @@
 pub struct HwServer {
     clients: IndexSlab<HwClient>,
     rooms: Slab<HwRoom>,
-    pub latest_protocol: u16,
-    pub flags: ServerFlags,
-    pub greetings: ServerGreetings,
+    latest_protocol: u16,
+    flags: ServerFlags,
+    greetings: ServerGreetings,
 }
 
 impl HwServer {
@@ -205,7 +205,7 @@
     }
 
     #[inline]
-    pub fn get_room_mut(&mut self, room_id: RoomId) -> Option<&mut HwRoom> {
+    fn get_room_mut(&mut self, room_id: RoomId) -> Option<&mut HwRoom> {
         self.rooms.get_mut(room_id)
     }
 
@@ -404,7 +404,7 @@
             .find_map(|(_, r)| Some(r).filter(|r| r.name == name))
     }
 
-    pub fn find_room_mut(&mut self, name: &str) -> Option<&mut HwRoom> {
+    fn find_room_mut(&mut self, name: &str) -> Option<&mut HwRoom> {
         self.rooms
             .iter_mut()
             .find_map(|(_, r)| Some(r).filter(|r| r.name == name))
@@ -416,13 +416,13 @@
             .find_map(|(_, c)| Some(c).filter(|c| c.nick == nick))
     }
 
-    pub fn find_client_mut(&mut self, nick: &str) -> Option<&mut HwClient> {
+    fn find_client_mut(&mut self, nick: &str) -> Option<&mut HwClient> {
         self.clients
             .iter_mut()
             .find_map(|(_, c)| Some(c).filter(|c| c.nick == nick))
     }
 
-    pub fn all_clients(&self) -> impl Iterator<Item = ClientId> + '_ {
+    pub fn iter_client_ids(&self) -> impl Iterator<Item = ClientId> + '_ {
         self.clients.iter().map(|(id, _)| id)
     }
 
@@ -440,7 +440,7 @@
         self.rooms.iter().filter(f).map(|(_, c)| c.id)
     }
 
-    pub fn collect_clients<F>(&self, f: F) -> Vec<ClientId>
+    pub fn collect_client_ids<F>(&self, f: F) -> Vec<ClientId>
     where
         F: Fn(&(usize, &HwClient)) -> bool,
     {
@@ -458,25 +458,25 @@
             .collect()
     }
 
-    pub fn lobby_clients(&self) -> impl Iterator<Item = ClientId> + '_ {
+    pub fn lobby_client_ids(&self) -> impl Iterator<Item = ClientId> + '_ {
         self.filter_clients(|(_, c)| c.room_id == None)
     }
 
-    pub fn room_clients(&self, room_id: RoomId) -> impl Iterator<Item = ClientId> + '_ {
+    pub fn room_client_ids(&self, room_id: RoomId) -> impl Iterator<Item = ClientId> + '_ {
         self.filter_clients(move |(_, c)| c.room_id == Some(room_id))
     }
 
-    pub fn protocol_clients(&self, protocol: u16) -> impl Iterator<Item = ClientId> + '_ {
+    pub fn protocol_client_ids(&self, protocol: u16) -> impl Iterator<Item = ClientId> + '_ {
         self.filter_clients(move |(_, c)| c.protocol_number == protocol)
     }
 
-    pub fn protocol_rooms(&self, protocol: u16) -> impl Iterator<Item = RoomId> + '_ {
+    pub fn protocol_room_ids(&self, protocol: u16) -> impl Iterator<Item = RoomId> + '_ {
         self.filter_rooms(move |(_, r)| r.protocol_number == protocol)
     }
 
-    pub fn other_clients_in_room(&self, self_id: ClientId) -> Vec<ClientId> {
+    pub fn other_client_ids_in_room(&self, self_id: ClientId) -> Vec<ClientId> {
         let room_id = self.clients[self_id].room_id;
-        self.collect_clients(|(id, c)| *id != self_id && c.room_id == room_id)
+        self.collect_client_ids(|(id, c)| *id != self_id && c.room_id == room_id)
     }
 
     pub fn is_registered_only(&self) -> bool {
@@ -486,6 +486,14 @@
     pub fn set_is_registered_only(&mut self, value: bool) {
         self.flags.set(ServerFlags::REGISTERED_ONLY, value)
     }
+
+    pub fn set_room_saves(&mut self, room_id: RoomId, text: &str) -> Result<(), serde_yaml::Error> {
+        if let Some(room) = self.rooms.get_mut(room_id) {
+            room.set_saves(text)
+        } else {
+            Ok(())
+        }
+    }
 }
 
 pub struct HwRoomControl<'a> {
@@ -590,7 +598,7 @@
                 self.server.rooms.remove(self.room_id);
             } else if room.master_id == None {
                 let protocol_number = room.protocol_number;
-                let new_master_id = self.server.room_clients(self.room_id).next();
+                let new_master_id = self.server.room_client_ids(self.room_id).next();
 
                 if let Some(new_master_id) = new_master_id {
                     let room = self.room_mut();
@@ -672,7 +680,7 @@
         match self.room().voting {
             Some(_) => Err(VotingInProgress),
             None => {
-                let voting = Voting::new(kind, self.server.room_clients(self.room_id).collect());
+                let voting = Voting::new(kind, self.server.room_client_ids(self.room_id).collect());
                 self.room_mut().voting = Some(voting);
                 Ok(())
             }
--- a/rust/hedgewars-server/src/handlers.rs	Tue Dec 24 20:57:58 2019 +0300
+++ b/rust/hedgewars-server/src/handlers.rs	Tue Dec 24 21:45:26 2019 +0300
@@ -233,10 +233,10 @@
         Destination::ToIds(ids) => ids,
         Destination::ToAll { group, skip_self } => {
             let mut ids: Vec<_> = match group {
-                DestinationGroup::All => server.all_clients().collect(),
-                DestinationGroup::Lobby => server.lobby_clients().collect(),
-                DestinationGroup::Protocol(proto) => server.protocol_clients(proto).collect(),
-                DestinationGroup::Room(id) => server.room_clients(id).collect(),
+                DestinationGroup::All => server.iter_client_ids().collect(),
+                DestinationGroup::Lobby => server.lobby_client_ids().collect(),
+                DestinationGroup::Protocol(proto) => server.protocol_client_ids(proto).collect(),
+                DestinationGroup::Room(id) => server.room_client_ids(id).collect(),
             };
 
             if skip_self {
@@ -460,13 +460,11 @@
             response.warn(ROOM_CONFIG_SAVE_FAILED);
         }
         IoResult::LoadRoom(room_id, Some(contents)) => {
-            if let Some(ref mut room) = state.server.get_room_mut(room_id) {
-                match room.set_saves(&contents) {
-                    Ok(_) => response.add(server_chat(ROOM_CONFIG_LOADED.to_string()).send_self()),
-                    Err(e) => {
-                        warn!("Error while deserializing the room configs: {}", e);
-                        response.warn(ROOM_CONFIG_DESERIALIZE_FAILED);
-                    }
+            match state.server.set_room_saves(room_id, &contents) {
+                Ok(_) => response.add(server_chat(ROOM_CONFIG_LOADED.to_string()).send_self()),
+                Err(e) => {
+                    warn!("Error while deserializing the room configs: {}", e);
+                    response.warn(ROOM_CONFIG_DESERIALIZE_FAILED);
                 }
             }
         }
--- a/rust/hedgewars-server/src/handlers/common.rs	Tue Dec 24 20:57:58 2019 +0300
+++ b/rust/hedgewars-server/src/handlers/common.rs	Tue Dec 24 21:45:26 2019 +0300
@@ -460,7 +460,7 @@
 
                     super::common::get_room_update(None, room, room_master, response);
 
-                    for client_id in room_control.server().room_clients(room.id) {
+                    for client_id in room_control.server().room_client_ids(room.id) {
                         super::common::get_room_config(room, client_id, response);
                     }
                 }
--- a/rust/hedgewars-server/src/handlers/inlobby.rs	Tue Dec 24 20:57:58 2019 +0300
+++ b/rust/hedgewars-server/src/handlers/inlobby.rs	Tue Dec 24 21:45:26 2019 +0300
@@ -103,8 +103,8 @@
                     html.push(format!(
                         "<tr><td>{}</td><td>{}</td><td>{}</td></tr>",
                         super::utils::protocol_version_string(protocol),
-                        server.protocol_clients(protocol).count(),
-                        server.protocol_rooms(protocol).count()
+                        server.protocol_client_ids(protocol).count(),
+                        server.protocol_room_ids(protocol).count()
                     ));
                 }
                 html.push("</table>".to_string());