slime/lazymc
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Grab exclusive lock for RCON invocations to server

Browse Source
This commit is contained in:
timvisee 2021-11-15 13:57:41 +01:00
parent 5ffc6ee911
commit 96d7fc9dec
No known key found for this signature in database
GPG Key ID: B8DB720BC383E172
1 changed files with 12 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
src/server.rs
View File

@ -6,6 +6,7 @@ use futures::FutureExt;
use minecraft_protocol::data::server_status::ServerStatus;
use tokio::process::Command;
use tokio::sync::watch;
use tokio::sync::Semaphore;
use tokio::time;
use crate::config::Config;
@ -99,6 +100,10 @@ pub struct Server {
/// Used as starting/stopping timeout.
kill_at: RwLock<Option<Instant>>,
/// Lock for exclusive RCON operations.
#[cfg(feature = "rcon")]
rcon_lock: Semaphore,
/// Last time server was stopped over RCON.
#[cfg(feature = "rcon")]
rcon_last_stop: Mutex<Option<Instant>>,
@ -342,6 +347,8 @@ impl Default for Server {
keep_online_until: Default::default(),
kill_at: Default::default(),
#[cfg(feature = "rcon")]
rcon_lock: Semaphore::new(1),
#[cfg(feature = "rcon")]
rcon_last_stop: Default::default(),
}
}
@ -425,6 +432,9 @@ async fn stop_server_rcon(config: &Config, server: &Server) -> bool {
return false;
}
// Grab RCON lock
let rcon_lock = server.rcon_lock.acquire().await.unwrap();
// Ensure RCON has cooled down
let rcon_cooled_down = server
.rcon_last_stop
@ -465,6 +475,8 @@ async fn stop_server_rcon(config: &Config, server: &Server) -> bool {
.replace(Instant::now());
server.update_state(State::Stopping, config);
drop(rcon_lock);
// Gracefully close connection
rcon.close().await;