#![deny(unsafe_code)]
#![allow(
clippy::option_map_unit_fn,
clippy::blocks_in_conditions,
clippy::needless_pass_by_ref_mut )]
#![deny(clippy::clone_on_ref_ptr)]
#![feature(
box_patterns,
let_chains,
never_type,
option_zip,
unwrap_infallible,
const_type_name
)]
pub mod automod;
mod character_creator;
pub mod chat;
pub mod chunk_generator;
mod chunk_serialize;
pub mod client;
pub mod cmd;
pub mod connection_handler;
mod data_dir;
pub mod error;
pub mod events;
pub mod input;
pub mod location;
pub mod lod;
pub mod login_provider;
pub mod metrics;
pub mod persistence;
mod pet;
pub mod presence;
pub mod rtsim;
pub mod settings;
pub mod state_ext;
pub mod sys;
#[cfg(feature = "persistent_world")]
pub mod terrain_persistence;
#[cfg(not(feature = "worldgen"))] mod test_world;
#[cfg(feature = "worldgen")] mod weather;
pub mod wiring;
pub use crate::{
data_dir::DEFAULT_DATA_DIR_NAME,
error::Error,
events::Event,
input::Input,
settings::{CalendarMode, EditableSettings, Settings},
};
#[cfg(feature = "persistent_world")]
use crate::terrain_persistence::TerrainPersistence;
use crate::{
automod::AutoMod,
chunk_generator::ChunkGenerator,
client::Client,
cmd::ChatCommandExt,
connection_handler::ConnectionHandler,
data_dir::DataDir,
location::Locations,
login_provider::LoginProvider,
persistence::PersistedComponents,
presence::{RegionSubscription, RepositionOnChunkLoad},
state_ext::StateExt,
sys::sentinel::DeletedEntities,
};
use censor::Censor;
#[cfg(not(feature = "worldgen"))]
use common::grid::Grid;
#[cfg(feature = "worldgen")]
use common::terrain::TerrainChunkSize;
use common::{
assets::AssetExt,
calendar::Calendar,
character::{CharacterId, CharacterItem},
cmd::ServerChatCommand,
comp,
event::{
register_event_busses, ClientDisconnectEvent, ClientDisconnectWithoutPersistenceEvent,
EventBus, ExitIngameEvent, UpdateCharacterDataEvent,
},
link::Is,
mounting::{Volume, VolumeRider},
region::RegionMap,
resources::{BattleMode, GameMode, Time, TimeOfDay},
rtsim::RtSimEntity,
shared_server_config::ServerConstants,
slowjob::SlowJobPool,
terrain::TerrainChunk,
util::GIT_DATE_TIMESTAMP,
vol::RectRasterableVol,
};
use common_base::prof_span;
use common_ecs::run_now;
use common_net::{
msg::{ClientType, DisconnectReason, ServerGeneral, ServerInfo, ServerMsg},
sync::WorldSyncExt,
};
use common_state::{AreasContainer, BlockDiff, BuildArea, State};
use common_systems::add_local_systems;
use metrics::{EcsSystemMetrics, PhysicsMetrics, TickMetrics};
use network::{ListenAddr, Network, Pid};
use persistence::{
character_loader::{CharacterLoader, CharacterUpdaterMessage},
character_updater::CharacterUpdater,
};
use prometheus::Registry;
use rustls::pki_types::{CertificateDer, PrivateKeyDer};
use specs::{
shred::SendDispatcher, Builder, Entity as EcsEntity, Entity, Join, LendJoin, WorldExt,
};
use std::{
ops::{Deref, DerefMut},
sync::{Arc, Mutex},
time::{Duration, Instant},
};
#[cfg(not(feature = "worldgen"))]
use test_world::{IndexOwned, World};
use tokio::runtime::Runtime;
use tracing::{debug, error, info, trace, warn};
use vek::*;
use veloren_query_server::server::QueryServer;
pub use world::{civ::WorldCivStage, sim::WorldSimStage, WorldGenerateStage};
use crate::{
persistence::{DatabaseSettings, SqlLogMode},
sys::terrain,
};
use hashbrown::HashMap;
use std::sync::RwLock;
use crate::settings::Protocol;
#[cfg(feature = "plugins")]
use {
common::uid::IdMaps,
common_state::plugin::{memory_manager::EcsWorld, PluginMgr},
};
use crate::{chat::ChatCache, persistence::character_loader::CharacterScreenResponseKind};
use common::comp::Anchor;
#[cfg(feature = "worldgen")]
pub use world::{
sim::{FileOpts, GenOpts, WorldOpts, DEFAULT_WORLD_MAP, DEFAULT_WORLD_SEED},
IndexOwned, World,
};
#[derive(Copy, Clone)]
pub struct SpawnPoint(pub Vec3<f32>);
impl Default for SpawnPoint {
fn default() -> Self { Self(Vec3::new(0.0, 0.0, 256.0)) }
}
pub const MIN_VD: u32 = 6;
#[derive(Copy, Clone, Default)]
pub struct Tick(u64);
#[derive(Clone)]
pub struct HwStats {
hardware_threads: u32,
rayon_threads: u32,
}
#[derive(Clone, Copy, PartialEq)]
enum DisconnectType {
WithPersistence,
WithoutPersistence,
}
#[derive(Copy, Clone)]
pub struct TickStart(Instant);
#[derive(Clone, Default, Debug)]
pub struct BattleModeBuffer {
map: HashMap<CharacterId, (BattleMode, Time)>,
}
impl BattleModeBuffer {
pub fn push(&mut self, char_id: CharacterId, save: (BattleMode, Time)) {
self.map.insert(char_id, save);
}
pub fn get(&self, char_id: &CharacterId) -> Option<&(BattleMode, Time)> {
self.map.get(char_id)
}
pub fn pop(&mut self, char_id: &CharacterId) -> Option<(BattleMode, Time)> {
self.map.remove(char_id)
}
}
pub struct ChunkRequest {
entity: EcsEntity,
key: Vec2<i32>,
}
#[derive(Debug)]
pub enum ServerInitStage {
DbMigrations,
DbVacuum,
WorldGen(WorldGenerateStage),
StartingSystems,
}
pub struct Server {
state: State,
world: Arc<World>,
index: IndexOwned,
connection_handler: ConnectionHandler,
runtime: Arc<Runtime>,
metrics_registry: Arc<Registry>,
chat_cache: ChatCache,
database_settings: Arc<RwLock<DatabaseSettings>>,
disconnect_all_clients_requested: bool,
server_constants: ServerConstants,
event_dispatcher: SendDispatcher<'static>,
}
impl Server {
pub fn new(
settings: Settings,
editable_settings: EditableSettings,
database_settings: DatabaseSettings,
data_dir: &std::path::Path,
report_stage: &(dyn Fn(ServerInitStage) + Send + Sync),
runtime: Arc<Runtime>,
) -> Result<Self, Error> {
prof_span!("Server::new");
info!("Server data dir is: {}", data_dir.display());
if settings.auth_server_address.is_none() {
info!("Authentication is disabled");
}
report_stage(ServerInitStage::DbMigrations);
debug!("Running DB migrations...");
persistence::run_migrations(&database_settings);
report_stage(ServerInitStage::DbVacuum);
debug!("Vacuuming database...");
persistence::vacuum_database(&database_settings);
let database_settings = Arc::new(RwLock::new(database_settings));
let registry = Arc::new(Registry::new());
let chunk_gen_metrics = metrics::ChunkGenMetrics::new(®istry).unwrap();
let job_metrics = metrics::JobMetrics::new(®istry).unwrap();
let network_request_metrics = metrics::NetworkRequestMetrics::new(®istry).unwrap();
let player_metrics = metrics::PlayerMetrics::new(®istry).unwrap();
let ecs_system_metrics = EcsSystemMetrics::new(®istry).unwrap();
let tick_metrics = TickMetrics::new(®istry).unwrap();
let physics_metrics = PhysicsMetrics::new(®istry).unwrap();
let server_event_metrics = metrics::ServerEventMetrics::new(®istry).unwrap();
let query_server_metrics = metrics::QueryServerMetrics::new(®istry).unwrap();
let battlemode_buffer = BattleModeBuffer::default();
let pools = State::pools(GameMode::Server);
#[cfg(feature = "plugins")]
let plugin_mgr = PluginMgr::from_asset_or_default();
debug!("Generating world, seed: {}", settings.world_seed);
#[cfg(feature = "worldgen")]
let (world, index) = World::generate(
settings.world_seed,
WorldOpts {
seed_elements: true,
world_file: if let Some(ref opts) = settings.map_file {
opts.clone()
} else {
FileOpts::LoadAsset(DEFAULT_WORLD_MAP.into())
},
calendar: Some(settings.calendar_mode.calendar_now()),
},
&pools,
&|stage| {
report_stage(ServerInitStage::WorldGen(stage));
},
);
#[cfg(not(feature = "worldgen"))]
let (world, index) = World::generate(settings.world_seed);
#[cfg(feature = "worldgen")]
let map = world.get_map_data(index.as_index_ref(), &pools);
#[cfg(not(feature = "worldgen"))]
let map = common_net::msg::WorldMapMsg {
dimensions_lg: Vec2::zero(),
max_height: 1.0,
rgba: Grid::new(Vec2::new(1, 1), 1),
horizons: [(vec![0], vec![0]), (vec![0], vec![0])],
alt: Grid::new(Vec2::new(1, 1), 1),
sites: Vec::new(),
possible_starting_sites: Vec::new(),
pois: Vec::new(),
default_chunk: Arc::new(world.generate_oob_chunk()),
};
#[cfg(feature = "worldgen")]
let map_size_lg = world.sim().map_size_lg();
#[cfg(not(feature = "worldgen"))]
let map_size_lg = world.map_size_lg();
let lod = lod::Lod::from_world(&world, index.as_index_ref(), &pools);
report_stage(ServerInitStage::StartingSystems);
let mut state = State::server(
Arc::clone(&pools),
map_size_lg,
Arc::clone(&map.default_chunk),
|dispatcher_builder| {
add_local_systems(dispatcher_builder);
sys::msg::add_server_systems(dispatcher_builder);
sys::add_server_systems(dispatcher_builder);
#[cfg(feature = "worldgen")]
{
rtsim::add_server_systems(dispatcher_builder);
weather::add_server_systems(dispatcher_builder);
}
},
#[cfg(feature = "plugins")]
plugin_mgr,
);
register_event_busses(state.ecs_mut());
state.ecs_mut().insert(battlemode_buffer);
state.ecs_mut().insert(settings.clone());
state.ecs_mut().insert(editable_settings);
state.ecs_mut().insert(DataDir {
path: data_dir.to_owned(),
});
register_event_busses(state.ecs_mut());
state.ecs_mut().insert(Vec::<ChunkRequest>::new());
state
.ecs_mut()
.insert(EventBus::<chunk_serialize::ChunkSendEntry>::default());
state.ecs_mut().insert(Locations::default());
state.ecs_mut().insert(LoginProvider::new(
settings.auth_server_address.clone(),
Arc::clone(&runtime),
));
state.ecs_mut().insert(HwStats {
hardware_threads: num_cpus::get() as u32,
rayon_threads: num_cpus::get() as u32,
});
state.ecs_mut().insert(Tick(0));
state.ecs_mut().insert(TickStart(Instant::now()));
state.ecs_mut().insert(job_metrics);
state.ecs_mut().insert(network_request_metrics);
state.ecs_mut().insert(player_metrics);
state.ecs_mut().insert(ecs_system_metrics);
state.ecs_mut().insert(tick_metrics);
state.ecs_mut().insert(physics_metrics);
state.ecs_mut().insert(server_event_metrics);
state.ecs_mut().insert(query_server_metrics);
if settings.experimental_terrain_persistence {
#[cfg(feature = "persistent_world")]
{
warn!(
"Experimental terrain persistence support is enabled. This feature may break, \
be disabled, or otherwise change under your feet at *any time*. \
Additionally, it is expected to be replaced in the future *without* \
migration or warning. You have been warned."
);
state
.ecs_mut()
.insert(TerrainPersistence::new(data_dir.to_owned()));
}
#[cfg(not(feature = "persistent_world"))]
error!(
"Experimental terrain persistence support was requested, but the server was not \
compiled with the feature. Terrain modifications will *not* be persisted."
);
}
{
let pool = state.ecs_mut().write_resource::<SlowJobPool>();
pool.configure("CHUNK_DROP", |_n| 1);
pool.configure("CHUNK_GENERATOR", |n| n / 2 + n / 4);
pool.configure("CHUNK_SERIALIZER", |n| n / 2);
pool.configure("RTSIM_SAVE", |_| 1);
pool.configure("WEATHER", |_| 1);
}
state
.ecs_mut()
.insert(ChunkGenerator::new(chunk_gen_metrics));
{
let (sender, receiver) =
crossbeam_channel::bounded::<chunk_serialize::SerializedChunk>(10_000);
state.ecs_mut().insert(sender);
state.ecs_mut().insert(receiver);
}
state.ecs_mut().insert(CharacterUpdater::new(
Arc::<RwLock<DatabaseSettings>>::clone(&database_settings),
)?);
let ability_map = comp::item::tool::AbilityMap::<comp::AbilityItem>::load_expect_cloned(
"common.abilities.ability_set_manifest",
);
state.ecs_mut().insert(ability_map);
let msm = comp::inventory::item::MaterialStatManifest::load().cloned();
state.ecs_mut().insert(msm);
let rbm = common::recipe::RecipeBookManifest::load().cloned();
state.ecs_mut().insert(rbm);
state.ecs_mut().insert(CharacterLoader::new(
Arc::<RwLock<DatabaseSettings>>::clone(&database_settings),
)?);
state
.ecs_mut()
.insert(sys::PersistenceScheduler::every(Duration::from_secs(10)));
state.ecs_mut().insert(RegionMap::new());
state.ecs_mut().register::<RegionSubscription>();
state.ecs_mut().register::<Client>();
state.ecs_mut().register::<comp::Presence>();
state.ecs_mut().register::<wiring::WiringElement>();
state.ecs_mut().register::<wiring::Circuit>();
state.ecs_mut().register::<Anchor>();
state.ecs_mut().register::<comp::Pet>();
state.ecs_mut().register::<login_provider::PendingLogin>();
state.ecs_mut().register::<RepositionOnChunkLoad>();
state.ecs_mut().register::<RtSimEntity>();
let banned_words = settings.moderation.load_banned_words(data_dir);
let censor = Arc::new(Censor::Custom(banned_words.into_iter().collect()));
state.ecs_mut().insert(Arc::clone(&censor));
state
.ecs_mut()
.insert(AutoMod::new(&settings.moderation, censor));
state.ecs_mut().insert(map);
#[cfg(feature = "worldgen")]
let spawn_point = SpawnPoint({
use world::civ::SiteKind;
let index = index.as_index_ref();
let center_chunk = world.sim().map_size_lg().chunks().map(i32::from) / 2;
let spawn_chunk = world
.civs()
.sites()
.filter(|site| matches!(site.kind, SiteKind::Settlement | SiteKind::Refactor))
.map(|site| site.center)
.min_by_key(|site_pos| site_pos.distance_squared(center_chunk))
.unwrap_or(center_chunk);
world.find_accessible_pos(index, TerrainChunkSize::center_wpos(spawn_chunk), false)
});
#[cfg(not(feature = "worldgen"))]
let spawn_point = SpawnPoint::default();
state.ecs_mut().insert(spawn_point);
{
#[cfg(feature = "worldgen")]
let size = world.sim().get_size();
#[cfg(not(feature = "worldgen"))]
let size = world.map_size_lg().chunks().map(u32::from);
let world_size = size.map(|e| e as i32) * TerrainChunk::RECT_SIZE.map(|e| e as i32);
let world_aabb = Aabb {
min: Vec3::new(0, 0, -32768),
max: Vec3::new(world_size.x, world_size.y, 32767),
}
.made_valid();
state
.ecs()
.write_resource::<AreasContainer<BuildArea>>()
.insert("world".to_string(), world_aabb)
.expect("The initial insert should always work.");
}
let world = Arc::new(world);
state.ecs_mut().insert(Arc::clone(&world));
state.ecs_mut().insert(lod);
state.ecs_mut().insert(index.clone());
state.ecs_mut().write_resource::<TimeOfDay>().0 = settings.world.start_time;
sys::sentinel::UpdateTrackers::register(state.ecs_mut());
state.ecs_mut().insert(DeletedEntities::default());
let network = Network::new_with_registry(Pid::new(), &runtime, ®istry);
let (chat_cache, chat_tracker) = ChatCache::new(Duration::from_secs(60), &runtime);
state.ecs_mut().insert(chat_tracker);
let mut printed_quic_warning = false;
for protocol in &settings.gameserver_protocols {
match protocol {
Protocol::Tcp { address } => {
runtime.block_on(network.listen(ListenAddr::Tcp(*address)))?;
},
Protocol::Quic {
address,
cert_file_path,
key_file_path,
} => {
use rustls_pemfile::Item;
use std::fs;
match || -> Result<_, Box<dyn std::error::Error>> {
let key = fs::read(key_file_path)?;
let key = if key_file_path.extension().map_or(false, |x| x == "der") {
PrivateKeyDer::try_from(key).map_err(|_| "No valid pem key in file")?
} else {
debug!("convert pem key to der");
rustls_pemfile::read_all(&mut key.as_slice())
.find_map(|item| match item {
Ok(Item::Pkcs1Key(v)) => Some(PrivateKeyDer::Pkcs1(v)),
Ok(Item::Pkcs8Key(v)) => Some(PrivateKeyDer::Pkcs8(v)),
Ok(Item::Sec1Key(v)) => Some(PrivateKeyDer::Sec1(v)),
Ok(Item::Crl(_)) => None,
Ok(Item::Csr(_)) => None,
Ok(Item::X509Certificate(_)) => None,
Ok(_) => None,
Err(e) => {
tracing::warn!(?e, "error while reading key_file");
None
},
})
.ok_or("No valid pem key in file")?
};
let cert_chain = fs::read(cert_file_path)?;
let cert_chain = if cert_file_path.extension().map_or(false, |x| x == "der")
{
vec![CertificateDer::from(cert_chain)]
} else {
debug!("convert pem cert to der");
rustls_pemfile::certs(&mut cert_chain.as_slice())
.filter_map(|item| match item {
Ok(cert) => Some(cert),
Err(e) => {
tracing::warn!(?e, "error while reading cert_file");
None
},
})
.collect()
};
let server_config = quinn::ServerConfig::with_single_cert(cert_chain, key)?;
Ok(server_config)
}() {
Ok(server_config) => {
runtime.block_on(
network.listen(ListenAddr::Quic(*address, server_config.clone())),
)?;
if !printed_quic_warning {
warn!(
"QUIC is enabled. This is experimental and not recommended in \
production"
);
printed_quic_warning = true;
}
},
Err(e) => {
error!(
?e,
"Failed to load the TLS certificate, running without QUIC {}",
*address
);
},
}
},
}
}
if let Some(addr) = settings.query_address {
use veloren_query_server::proto::ServerInfo;
const QUERY_SERVER_RATELIMIT: u16 = 120;
let (query_server_info_tx, query_server_info_rx) =
tokio::sync::watch::channel(ServerInfo {
git_hash: *sys::server_info::GIT_HASH,
git_timestamp: *GIT_DATE_TIMESTAMP,
players_count: 0,
player_cap: settings.max_players,
battlemode: settings.gameplay.battle_mode.into(),
});
let mut query_server =
QueryServer::new(addr, query_server_info_rx, QUERY_SERVER_RATELIMIT);
let query_server_metrics =
Arc::new(Mutex::new(veloren_query_server::server::Metrics::default()));
let query_server_metrics2 = Arc::clone(&query_server_metrics);
runtime.spawn(async move {
let err = query_server.run(query_server_metrics2).await.err();
error!(?err, "Query server stopped unexpectedly");
});
state.ecs_mut().insert(query_server_info_tx);
state.ecs_mut().insert(query_server_metrics);
}
runtime.block_on(network.listen(ListenAddr::Mpsc(14004)))?;
let connection_handler = ConnectionHandler::new(network, &runtime);
#[cfg(feature = "worldgen")]
{
match rtsim::RtSim::new(
&settings.world,
index.as_index_ref(),
&world,
data_dir.to_owned(),
) {
Ok(rtsim) => {
state.ecs_mut().insert(rtsim.state().data().time_of_day);
state.ecs_mut().insert(rtsim);
},
Err(err) => {
error!("Failed to load rtsim: {}", err);
return Err(Error::RtsimError(err));
},
}
weather::init(&mut state);
}
let server_constants = ServerConstants {
day_cycle_coefficient: settings.day_cycle_coefficient(),
};
let this = Self {
state,
world,
index,
connection_handler,
runtime,
metrics_registry: registry,
chat_cache,
database_settings,
disconnect_all_clients_requested: false,
server_constants,
event_dispatcher: Self::create_event_dispatcher(pools),
};
debug!(?settings, "created veloren server with");
let git_hash = *common::util::GIT_HASH;
let git_date = common::util::GIT_DATE.clone();
let git_time = *common::util::GIT_TIME;
let version = common::util::DISPLAY_VERSION_LONG.clone();
info!(?version, "Server version");
debug!(?git_hash, ?git_date, ?git_time, "detailed Server version");
Ok(this)
}
pub fn get_server_info(&self) -> ServerInfo {
let settings = self.state.ecs().fetch::<Settings>();
ServerInfo {
name: settings.server_name.clone(),
git_hash: common::util::GIT_HASH.to_string(),
git_date: common::util::GIT_DATE.to_string(),
auth_provider: settings.auth_server_address.clone(),
}
}
pub fn settings(&self) -> impl Deref<Target = Settings> + '_ {
self.state.ecs().fetch::<Settings>()
}
pub fn settings_mut(&self) -> impl DerefMut<Target = Settings> + '_ {
self.state.ecs().fetch_mut::<Settings>()
}
pub fn editable_settings_mut(&self) -> impl DerefMut<Target = EditableSettings> + '_ {
self.state.ecs().fetch_mut::<EditableSettings>()
}
pub fn editable_settings(&self) -> impl Deref<Target = EditableSettings> + '_ {
self.state.ecs().fetch::<EditableSettings>()
}
pub fn data_dir(&self) -> impl Deref<Target = DataDir> + '_ {
self.state.ecs().fetch::<DataDir>()
}
pub fn state(&self) -> &State { &self.state }
pub fn state_mut(&mut self) -> &mut State { &mut self.state }
pub fn world(&self) -> &World { &self.world }
pub fn metrics_registry(&self) -> &Arc<Registry> { &self.metrics_registry }
pub fn chat_cache(&self) -> &ChatCache { &self.chat_cache }
fn parse_locations(&self, character_list_data: &mut [CharacterItem]) {
character_list_data.iter_mut().for_each(|c| {
let name = c
.location
.as_ref()
.and_then(|s| {
persistence::parse_waypoint(s)
.ok()
.and_then(|(waypoint, _)| waypoint.map(|w| w.get_pos()))
})
.and_then(|wpos| {
self.world
.get_location_name(self.index.as_index_ref(), wpos.xy().as_::<i32>())
});
c.location = name;
});
}
pub fn tick(&mut self, _input: Input, dt: Duration) -> Result<Vec<Event>, Error> {
self.state.ecs().write_resource::<Tick>().0 += 1;
self.state.ecs().write_resource::<TickStart>().0 = Instant::now();
let new_calendar = self
.state
.ecs()
.read_resource::<Settings>()
.calendar_mode
.calendar_now();
*self.state.ecs_mut().write_resource::<Calendar>() = new_calendar;
let mut frontend_events = Vec::new();
let before_new_connections = Instant::now();
self.handle_new_connections(&mut frontend_events);
let before_state_tick = Instant::now();
fn on_block_update(ecs: &specs::World, changes: Vec<BlockDiff>) {
if changes
.iter()
.any(|c| c.old.get_rtsim_resource() != c.new.get_rtsim_resource())
{
ecs.write_resource::<rtsim::RtSim>().hook_block_update(
&ecs.read_resource::<Arc<world::World>>(),
ecs.read_resource::<world::IndexOwned>().as_index_ref(),
changes,
);
}
}
let mut state_tick_metrics = Default::default();
self.state.tick(
dt,
false,
Some(&mut state_tick_metrics),
&self.server_constants,
on_block_update,
);
let before_handle_events = Instant::now();
let disconnect_type = self.disconnect_all_clients_if_requested();
self.state.maintain_links();
frontend_events.append(&mut self.handle_events());
let before_update_terrain_and_regions = Instant::now();
self.update_region_map();
self.state.apply_terrain_changes(on_block_update);
let before_sync = Instant::now();
sys::run_sync_systems(self.state.ecs_mut());
let before_world_tick = Instant::now();
self.world.tick(dt);
let before_entity_cleanup = Instant::now();
if let Some(DisconnectType::WithoutPersistence) = disconnect_type {
run_now::<terrain::Sys>(self.state.ecs_mut());
}
#[cfg(feature = "worldgen")]
{
let mut rtsim = self.state.ecs().write_resource::<rtsim::RtSim>();
for chunk in &self.state.terrain_changes().removed_chunks {
rtsim.hook_unload_chunk(*chunk);
}
}
let anchors = self.state.ecs().read_storage::<Anchor>();
let anchored_anchor_entities: Vec<Entity> = (
&self.state.ecs().entities(),
&self.state.ecs().read_storage::<Anchor>(),
)
.join()
.filter_map(|(_, anchor)| match anchor {
Anchor::Entity(anchor_entity) => Some(*anchor_entity),
_ => None,
})
.filter(|anchor_entity| match anchors.get(*anchor_entity) {
Some(Anchor::Entity(_)) => true,
Some(Anchor::Chunk(_)) | None => false
})
.collect();
drop(anchors);
for entity in anchored_anchor_entities {
if cfg!(debug_assertions) {
panic!("Entity anchor chain detected");
}
error!(
"Detected an anchor entity that itself has an anchor entity - anchor chains are \
not currently supported. The entity's Anchor component has been deleted"
);
self.state.delete_component::<Anchor>(entity);
}
let to_delete = {
let terrain = self.state.terrain();
(
&self.state.ecs().entities(),
&self.state.ecs().read_storage::<comp::Pos>(),
!&self.state.ecs().read_storage::<comp::Presence>(),
self.state.ecs().read_storage::<Anchor>().maybe(),
self.state.ecs().read_storage::<Is<VolumeRider>>().maybe(),
)
.join()
.filter(|(_, pos, _, anchor, is_volume_rider)| {
let pos = is_volume_rider
.and_then(|is_volume_rider| match is_volume_rider.pos.kind {
Volume::Terrain => None,
Volume::Entity(e) => {
let e = self.state.ecs().entity_from_uid(e)?;
let pos = self
.state
.ecs()
.read_storage::<comp::Pos>()
.get(e)
.copied()?;
Some(pos.0)
},
})
.unwrap_or(pos.0);
let chunk_key = terrain.pos_key(pos.map(|e| e.floor() as i32));
match anchor {
Some(Anchor::Chunk(hc)) => {
terrain.get_key_real(chunk_key).is_none()
&& terrain.get_key_real(*hc).is_none()
},
Some(Anchor::Entity(entity)) => !self.state.ecs().is_alive(*entity),
None => terrain.get_key_real(chunk_key).is_none(),
}
})
.map(|(entity, _, _, _, _)| entity)
.collect::<Vec<_>>()
};
#[cfg(feature = "worldgen")]
{
let mut rtsim = self.state.ecs().write_resource::<rtsim::RtSim>();
let rtsim_entities = self.state.ecs().read_storage();
for entity in &to_delete {
if let Some(rtsim_entity) = rtsim_entities.get(*entity) {
rtsim.hook_rtsim_entity_unload(*rtsim_entity);
}
}
}
for entity in to_delete {
if let Err(e) = self.state.delete_entity_recorded(entity) {
error!(?e, "Failed to delete agent outside the terrain");
}
}
if let Some(DisconnectType::WithoutPersistence) = disconnect_type {
info!(
"Disconnection of all players without persistence complete, signalling to \
persistence thread that character updates may continue to be processed"
);
self.state
.ecs()
.fetch_mut::<CharacterUpdater>()
.disconnected_success();
}
let before_persistence_updates = Instant::now();
let character_loader = self.state.ecs().read_resource::<CharacterLoader>();
let mut character_updater = self.state.ecs().write_resource::<CharacterUpdater>();
let updater_messages: Vec<CharacterUpdaterMessage> = character_updater.messages().collect();
character_loader
.messages()
.chain(updater_messages)
.for_each(|message| match message {
CharacterUpdaterMessage::DatabaseBatchCompletion(batch_id) => {
character_updater.process_batch_completion(batch_id);
},
CharacterUpdaterMessage::CharacterScreenResponse(response) => {
match response.response_kind {
CharacterScreenResponseKind::CharacterList(result) => match result {
Ok(mut character_list_data) => {
self.parse_locations(&mut character_list_data);
self.notify_client(
response.target_entity,
ServerGeneral::CharacterListUpdate(character_list_data),
)
},
Err(error) => self.notify_client(
response.target_entity,
ServerGeneral::CharacterActionError(error.to_string()),
),
},
CharacterScreenResponseKind::CharacterCreation(result) => match result {
Ok((character_id, mut list)) => {
self.parse_locations(&mut list);
self.notify_client(
response.target_entity,
ServerGeneral::CharacterListUpdate(list),
);
self.notify_client(
response.target_entity,
ServerGeneral::CharacterCreated(character_id),
);
},
Err(error) => self.notify_client(
response.target_entity,
ServerGeneral::CharacterActionError(error.to_string()),
),
},
CharacterScreenResponseKind::CharacterEdit(result) => match result {
Ok((character_id, mut list)) => {
self.parse_locations(&mut list);
self.notify_client(
response.target_entity,
ServerGeneral::CharacterListUpdate(list),
);
self.notify_client(
response.target_entity,
ServerGeneral::CharacterEdited(character_id),
);
},
Err(error) => self.notify_client(
response.target_entity,
ServerGeneral::CharacterActionError(error.to_string()),
),
},
CharacterScreenResponseKind::CharacterData(result) => {
match *result {
Ok((character_data, skill_set_persistence_load_error)) => {
let PersistedComponents {
body,
hardcore,
stats,
skill_set,
inventory,
waypoint,
pets,
active_abilities,
map_marker,
} = character_data;
let character_data = (
body,
hardcore,
stats,
skill_set,
inventory,
waypoint,
pets,
active_abilities,
map_marker,
);
self.state.emit_event_now(UpdateCharacterDataEvent {
entity: response.target_entity,
components: character_data,
metadata: skill_set_persistence_load_error,
})
},
Err(error) => {
self.notify_client(
response.target_entity,
ServerGeneral::CharacterDataLoadResult(Err(
error.to_string()
)),
);
self.state.emit_event_now(ExitIngameEvent {
entity: response.target_entity,
})
},
}
},
}
},
});
drop(character_loader);
drop(character_updater);
{
let index = &mut self.index;
let world = &mut self.world;
let ecs = self.state.ecs_mut();
let slow_jobs = ecs.write_resource::<SlowJobPool>();
index.reload_if_changed(|index| {
let mut chunk_generator = ecs.write_resource::<ChunkGenerator>();
let client = ecs.read_storage::<Client>();
let mut terrain = ecs.write_resource::<common::terrain::TerrainGrid>();
#[cfg(feature = "worldgen")]
let rtsim = ecs.read_resource::<rtsim::RtSim>();
#[cfg(not(feature = "worldgen"))]
let rtsim = ();
chunk_generator.cancel_all();
if client.is_empty() {
terrain.clear();
} else {
terrain.iter().for_each(|(pos, _)| {
chunk_generator.generate_chunk(
None,
pos,
&slow_jobs,
Arc::clone(world),
&rtsim,
index.clone(),
(
*ecs.read_resource::<TimeOfDay>(),
(*ecs.read_resource::<Calendar>()).clone(),
),
);
});
}
});
}
let end_of_server_tick = Instant::now();
run_now::<sys::metrics::Sys>(self.state.ecs());
{
let tick_metrics = self.state.ecs().read_resource::<TickMetrics>();
let tt = &tick_metrics.tick_time;
tt.with_label_values(&["new connections"])
.set((before_state_tick - before_new_connections).as_nanos() as i64);
tt.with_label_values(&["handle server events"])
.set((before_update_terrain_and_regions - before_handle_events).as_nanos() as i64);
tt.with_label_values(&["update terrain and region map"])
.set((before_sync - before_update_terrain_and_regions).as_nanos() as i64);
tt.with_label_values(&["state"])
.set((before_handle_events - before_state_tick).as_nanos() as i64);
tt.with_label_values(&["world tick"])
.set((before_entity_cleanup - before_world_tick).as_nanos() as i64);
tt.with_label_values(&["entity cleanup"])
.set((before_persistence_updates - before_entity_cleanup).as_nanos() as i64);
tt.with_label_values(&["persistence_updates"])
.set((end_of_server_tick - before_persistence_updates).as_nanos() as i64);
for (label, duration) in state_tick_metrics.timings {
tick_metrics
.state_tick_time
.with_label_values(&[label])
.set(duration.as_nanos() as i64);
}
tick_metrics.tick_time_hist.observe(
end_of_server_tick
.duration_since(before_state_tick)
.as_secs_f64(),
);
}
Ok(frontend_events)
}
pub fn cleanup(&mut self) {
self.state.cleanup();
#[cfg(feature = "persistent_world")]
self.state
.ecs()
.try_fetch_mut::<TerrainPersistence>()
.map(|mut t| t.maintain());
}
fn update_region_map(&mut self) {
prof_span!("Server::update_region_map");
let ecs = self.state().ecs();
ecs.write_resource::<RegionMap>().tick(
ecs.read_storage::<comp::Pos>(),
ecs.read_storage::<comp::Vel>(),
ecs.read_storage::<comp::Presence>(),
ecs.entities(),
);
}
fn initialize_client(&mut self, client: connection_handler::IncomingClient) -> Entity {
let entity = self
.state
.ecs_mut()
.create_entity_synced()
.with(client)
.build();
self.state
.ecs()
.read_resource::<metrics::PlayerMetrics>()
.clients_connected
.inc();
entity
}
fn disconnect_all_clients_if_requested(&mut self) -> Option<DisconnectType> {
let mut character_updater = self.state.ecs().fetch_mut::<CharacterUpdater>();
let disconnect_type = self.get_disconnect_all_clients_requested(&mut character_updater);
if let Some(disconnect_type) = disconnect_type {
let with_persistence = disconnect_type == DisconnectType::WithPersistence;
let clients = self.state.ecs().read_storage::<Client>();
let entities = self.state.ecs().entities();
info!(
"Disconnecting all clients ({} persistence) as requested",
if with_persistence { "with" } else { "without" }
);
for (_, entity) in (&clients, &entities).join() {
info!("Emitting client disconnect event for entity: {:?}", entity);
if with_persistence {
self.state.emit_event_now(ClientDisconnectEvent(
entity,
comp::DisconnectReason::Kicked,
))
} else {
self.state
.emit_event_now(ClientDisconnectWithoutPersistenceEvent(entity))
};
}
self.disconnect_all_clients_requested = false;
}
disconnect_type
}
fn get_disconnect_all_clients_requested(
&self,
character_updater: &mut CharacterUpdater,
) -> Option<DisconnectType> {
let without_persistence_requested = character_updater.disconnect_all_clients_requested();
let with_persistence_requested = self.disconnect_all_clients_requested;
if without_persistence_requested {
return Some(DisconnectType::WithoutPersistence);
};
if with_persistence_requested {
return Some(DisconnectType::WithPersistence);
};
None
}
fn handle_new_connections(&mut self, frontend_events: &mut Vec<Event>) {
while let Ok(sender) = self.connection_handler.info_requester_receiver.try_recv() {
trace!("sending info to connection_handler");
let _ = sender.send(connection_handler::ServerInfoPacket {
info: self.get_server_info(),
time: self.state.get_time(),
});
}
while let Ok(incoming) = self.connection_handler.client_receiver.try_recv() {
let entity = self.initialize_client(incoming);
frontend_events.push(Event::ClientConnected { entity });
}
}
pub fn notify_client<S>(&self, entity: EcsEntity, msg: S)
where
S: Into<ServerMsg>,
{
if let Some(client) = self.state.ecs().read_storage::<Client>().get(entity) {
client.send_fallible(msg);
}
}
pub fn notify_players(&mut self, msg: ServerGeneral) { self.state.notify_players(msg); }
pub fn generate_chunk(&mut self, entity: EcsEntity, key: Vec2<i32>) {
let ecs = self.state.ecs();
let slow_jobs = ecs.read_resource::<SlowJobPool>();
#[cfg(feature = "worldgen")]
let rtsim = ecs.read_resource::<rtsim::RtSim>();
#[cfg(not(feature = "worldgen"))]
let rtsim = ();
ecs.write_resource::<ChunkGenerator>().generate_chunk(
Some(entity),
key,
&slow_jobs,
Arc::clone(&self.world),
&rtsim,
self.index.clone(),
(
*ecs.read_resource::<TimeOfDay>(),
(*ecs.read_resource::<Calendar>()).clone(),
),
);
}
fn process_command(&mut self, entity: EcsEntity, name: String, args: Vec<String>) {
if let Ok(command) = name.parse::<ServerChatCommand>() {
command.execute(self, entity, args);
} else {
#[cfg(feature = "plugins")]
{
let mut plugin_manager = self.state.ecs().write_resource::<PluginMgr>();
let ecs_world = EcsWorld {
entities: &self.state.ecs().entities(),
health: self.state.ecs().read_component().into(),
uid: self.state.ecs().read_component().into(),
id_maps: &self.state.ecs().read_resource::<IdMaps>().into(),
player: self.state.ecs().read_component().into(),
};
let uid = if let Some(uid) = ecs_world.uid.get(entity).copied() {
uid
} else {
self.notify_client(
entity,
ServerGeneral::server_msg(
comp::ChatType::CommandError,
common::comp::Content::Plain(
"Can't get player UUID (player may be disconnected?)".to_string(),
),
),
);
return;
};
match plugin_manager.command_event(&ecs_world, &name, args.as_slice(), uid) {
Err(common_state::plugin::CommandResults::UnknownCommand) => self
.notify_client(
entity,
ServerGeneral::server_msg(
comp::ChatType::CommandError,
common::comp::Content::Plain(format!(
"Unknown command '/{name}'.\nType '/help' for available \
commands",
)),
),
),
Ok(value) => {
self.notify_client(
entity,
ServerGeneral::server_msg(
comp::ChatType::CommandInfo,
common::comp::Content::Plain(value.join("\n")),
),
);
},
Err(common_state::plugin::CommandResults::PluginError(err)) => {
self.notify_client(
entity,
ServerGeneral::server_msg(
comp::ChatType::CommandError,
common::comp::Content::Plain(format!(
"Error occurred while executing command '/{name}'.\n{err}"
)),
),
);
},
Err(common_state::plugin::CommandResults::HostError(err)) => {
error!(?err, ?name, ?args, "Can't execute command");
self.notify_client(
entity,
ServerGeneral::server_msg(
comp::ChatType::CommandError,
common::comp::Content::Plain(format!(
"Internal error {err:?} while executing '/{name}'.\nContact \
the server administrator",
)),
),
);
},
}
}
}
}
fn entity_admin_role(&self, entity: EcsEntity) -> Option<comp::AdminRole> {
self.state
.read_component_copied::<comp::Admin>(entity)
.map(|admin| admin.0)
}
pub fn number_of_players(&self) -> i64 {
self.state.ecs().read_storage::<Client>().join().count() as i64
}
pub fn add_admin(&mut self, username: &str, role: comp::AdminRole) {
let mut editable_settings = self.editable_settings_mut();
let login_provider = self.state.ecs().fetch::<LoginProvider>();
let data_dir = self.data_dir();
if let Some(entity) = add_admin(
username,
role,
&login_provider,
&mut editable_settings,
&data_dir.path,
)
.and_then(|uuid| {
let state = &self.state;
(
&state.ecs().entities(),
&state.read_storage::<comp::Player>(),
)
.join()
.find(|(_, player)| player.uuid() == uuid)
.map(|(e, _)| e)
}) {
drop((data_dir, login_provider, editable_settings));
self.state
.write_component_ignore_entity_dead(entity, comp::Admin(role));
};
}
pub fn remove_admin(&self, username: &str) {
let mut editable_settings = self.editable_settings_mut();
let login_provider = self.state.ecs().fetch::<LoginProvider>();
let data_dir = self.data_dir();
if let Some(entity) = remove_admin(
username,
&login_provider,
&mut editable_settings,
&data_dir.path,
)
.and_then(|uuid| {
let state = &self.state;
(
&state.ecs().entities(),
&state.read_storage::<comp::Player>(),
)
.join()
.find(|(_, player)| player.uuid() == uuid)
.map(|(e, _)| e)
}) {
self.state
.ecs()
.write_storage::<comp::Admin>()
.remove(entity);
};
}
#[cfg(feature = "worldgen")]
pub fn create_centered_persister(&mut self, view_distance: u32) {
let world_dims_chunks = self.world.sim().get_size();
let world_dims_blocks = TerrainChunkSize::blocks(world_dims_chunks);
let pos = comp::Pos(Vec3::from(world_dims_blocks.map(|e| e as f32 / 2.0)));
self.state
.create_persister(pos, view_distance, &self.world, &self.index)
.build();
}
pub fn chunks_pending(&mut self) -> bool {
self.state_mut()
.mut_resource::<ChunkGenerator>()
.pending_chunks()
.next()
.is_some()
}
pub fn set_sql_log_mode(&mut self, sql_log_mode: SqlLogMode) {
let mut database_settings = self.database_settings.write().unwrap();
database_settings.sql_log_mode = sql_log_mode;
drop(database_settings);
info!("SQL log mode changed to {:?}", sql_log_mode);
}
pub fn disconnect_all_clients(&mut self) {
info!("Disconnecting all clients due to local console command");
self.disconnect_all_clients_requested = true;
}
}
impl Drop for Server {
fn drop(&mut self) {
self.state
.notify_players(ServerGeneral::Disconnect(DisconnectReason::Shutdown));
#[cfg(feature = "persistent_world")]
self.state
.ecs()
.try_fetch_mut::<TerrainPersistence>()
.map(|mut terrain_persistence| {
info!("Unloading terrain persistence...");
terrain_persistence.unload_all()
});
#[cfg(feature = "worldgen")]
{
debug!("Saving rtsim state...");
self.state.ecs().write_resource::<rtsim::RtSim>().save(true);
}
}
}
#[must_use]
pub fn handle_edit<T, S: settings::EditableSetting>(
data: T,
result: Option<(String, Result<(), settings::SettingError<S>>)>,
) -> Option<T> {
use crate::settings::SettingError;
let (info, result) = result?;
match result {
Ok(()) => {
info!("{}", info);
Some(data)
},
Err(SettingError::Io(err)) => {
warn!(
?err,
"Failed to write settings file to disk, but succeeded in memory (success message: \
{})",
info,
);
Some(data)
},
Err(SettingError::Integrity(err)) => {
error!(?err, "Encountered an error while validating the request",);
None
},
}
}
#[must_use]
pub fn add_admin(
username: &str,
role: comp::AdminRole,
login_provider: &LoginProvider,
editable_settings: &mut EditableSettings,
data_dir: &std::path::Path,
) -> Option<common::uuid::Uuid> {
use crate::settings::EditableSetting;
let role_ = role.into();
match login_provider.username_to_uuid(username) {
Ok(uuid) => handle_edit(
uuid,
editable_settings.admins.edit(data_dir, |admins| {
match admins.insert(uuid, settings::AdminRecord {
username_when_admined: Some(username.into()),
date: chrono::Utc::now(),
role: role_,
}) {
None => Some(format!(
"Successfully added {} ({}) as {:?}!",
username, uuid, role
)),
Some(old_admin) if old_admin.role == role_ => {
info!("{} ({}) already has role: {:?}!", username, uuid, role);
None
},
Some(old_admin) => Some(format!(
"{} ({}) role changed from {:?} to {:?}!",
username, uuid, old_admin.role, role
)),
}
}),
),
Err(err) => {
error!(
?err,
"Could not find uuid for this name; either the user does not exist or there was \
an error communicating with the auth server."
);
None
},
}
}
#[must_use]
pub fn remove_admin(
username: &str,
login_provider: &LoginProvider,
editable_settings: &mut EditableSettings,
data_dir: &std::path::Path,
) -> Option<common::uuid::Uuid> {
use crate::settings::EditableSetting;
match login_provider.username_to_uuid(username) {
Ok(uuid) => handle_edit(
uuid,
editable_settings.admins.edit(data_dir, |admins| {
if let Some(admin) = admins.remove(&uuid) {
Some(format!(
"Successfully removed {} ({}) with role {:?} from the admins list",
username, uuid, admin.role,
))
} else {
info!("{} ({}) is not an admin!", username, uuid);
None
}
}),
),
Err(err) => {
error!(
?err,
"Could not find uuid for this name; either the user does not exist or there was \
an error communicating with the auth server."
);
None
},
}
}