mercuryworkshop-mirror/epoxy-tls
1
0
Fork 0
mirror of https://github.com/MercuryWorkshop/epoxy-tls.git synced 2025-05-12 14:00:01 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions

hide everything except config from rustdoc

Browse source
This commit is contained in:
Toshit Chawda 2024-11-22 22:28:22 -08:00
parent a554a5a761
commit 1128ded1bb
No known key found for this signature in database
GPG key ID: 91480ED99E2B3D9D
2 changed files with 40 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

27
server/src/config.rs
View file

@ -19,7 +19,7 @@ use wisp_mux::{
use crate::{handle::wisp::utils::get_certificates_from_paths, CLI, CONFIG, RESOLVER};
const VERSION_STRING: &str = concat!(
pub const VERSION_STRING: &str = concat!(
"git ",
env!("VERGEN_GIT_SHA"),
", dirty ",
@ -134,10 +134,12 @@ pub enum ProtocolExtensionAuth {
Certificate,
}
#[doc(hidden)]
fn default_motd() -> String {
format!("epoxy_server ({})", VERSION_STRING)
}
#[doc(hidden)]
fn is_default_motd(str: &String) -> bool {
*str == default_motd()
}
@ -160,10 +162,10 @@ pub struct WispConfig {
pub auth_extension: Option<ProtocolExtensionAuth>,
#[cfg(feature = "speed-limit")]
/// Read limit in bytes/second for all streams.
/// Read limit in bytes/second for all streams in a wisp connection.
pub read_limit: f64,
#[cfg(feature = "speed-limit")]
/// Write limit in bytes/second for all streams.
/// Write limit in bytes/second for all streams in a wisp connection.
pub write_limit: f64,
#[serde(skip_serializing_if = "HashMap::is_empty")]
@ -235,11 +237,15 @@ pub struct StreamConfig {
#[derive(Debug, Serialize, Deserialize, Default)]
#[serde(default)]
pub struct Config {
/// Server-specific config.
pub server: ServerConfig,
/// Wisp-specific configuration.
pub wisp: WispConfig,
/// Individual stream-specific configuration.
pub stream: StreamConfig,
}
#[doc(hidden)]
#[derive(Debug)]
struct ConfigCache {
pub blocked_ports: Vec<RangeInclusive<u16>>,
@ -256,6 +262,7 @@ struct ConfigCache {
}
lazy_static! {
#[doc(hidden)]
static ref CONFIG_CACHE: ConfigCache = {
ConfigCache {
allowed_ports: CONFIG
@ -283,6 +290,7 @@ lazy_static! {
};
}
#[doc(hidden)]
pub async fn validate_config_cache() {
// constructs regexes
let _ = CONFIG_CACHE.allowed_ports;
@ -358,6 +366,7 @@ impl Default for WispConfig {
}
impl WispConfig {
#[doc(hidden)]
pub async fn to_opts(&self) -> anyhow::Result<(Option<WispV2Handshake>, Vec<u8>, u32)> {
if self.wisp_v2 {
let mut extensions: Vec<AnyProtocolExtensionBuilder> = Vec::new();
@ -449,40 +458,49 @@ impl Default for StreamConfig {
}
impl StreamConfig {
#[doc(hidden)]
pub fn allowed_ports(&self) -> &'static [RangeInclusive<u16>] {
&CONFIG_CACHE.allowed_ports
}
#[doc(hidden)]
pub fn blocked_ports(&self) -> &'static [RangeInclusive<u16>] {
&CONFIG_CACHE.blocked_ports
}
#[doc(hidden)]
pub fn allowed_hosts(&self) -> &RegexSet {
&CONFIG_CACHE.allowed_hosts
}
#[doc(hidden)]
pub fn blocked_hosts(&self) -> &RegexSet {
&CONFIG_CACHE.blocked_hosts
}
#[doc(hidden)]
pub fn allowed_tcp_hosts(&self) -> &RegexSet {
&CONFIG_CACHE.allowed_tcp_hosts
}
#[doc(hidden)]
pub fn blocked_tcp_hosts(&self) -> &RegexSet {
&CONFIG_CACHE.blocked_tcp_hosts
}
#[doc(hidden)]
pub fn allowed_udp_hosts(&self) -> &RegexSet {
&CONFIG_CACHE.allowed_udp_hosts
}
#[doc(hidden)]
pub fn blocked_udp_hosts(&self) -> &RegexSet {
&CONFIG_CACHE.blocked_udp_hosts
}
}
impl Config {
#[doc(hidden)]
pub fn ser(&self) -> anyhow::Result<String> {
Ok(match CLI.format {
ConfigFormat::Json => serde_json::to_string_pretty(self)?,
@ -493,6 +511,7 @@ impl Config {
})
}
#[doc(hidden)]
pub fn de(string: String) -> anyhow::Result<Self> {
Ok(match CLI.format {
ConfigFormat::Json => serde_json::from_str(&string)?,
@ -505,6 +524,7 @@ impl Config {
}
#[derive(Debug, Clone, Copy, Eq, PartialEq, ValueEnum)]
#[doc(hidden)]
pub enum ConfigFormat {
Json,
#[cfg(feature = "toml")]
@ -528,6 +548,7 @@ impl Default for ConfigFormat {
}
/// Performant server implementation of the Wisp protocol in Rust, made for epoxy.
#[doc(hidden)]
#[derive(Parser, Debug)]
#[command(version = VERSION_STRING)]
pub struct Cli {

18
server/src/main.rs
View file

@ -1,4 +1,4 @@
#![feature(ip)]
#![doc(html_no_source)]
#![deny(clippy::todo)]
#![allow(unexpected_cfgs)]
@ -26,15 +26,22 @@ use tokio::{
use uuid::Uuid;
use wisp_mux::ConnectPacket;
mod config;
pub mod config;
#[doc(hidden)]
mod handle;
#[doc(hidden)]
mod listener;
#[doc(hidden)]
mod route;
#[doc(hidden)]
mod stats;
#[doc(hidden)]
mod stream;
#[doc(hidden)]
type Client = (DashMap<Uuid, (ConnectPacket, ConnectPacket)>, bool);
#[doc(hidden)]
#[derive(Debug)]
pub enum Resolver {
Hickory(TokioAsyncResolver),
@ -60,7 +67,9 @@ impl Resolver {
}
lazy_static! {
#[doc(hidden)]
pub static ref CLI: Cli = Cli::parse();
#[doc(hidden)]
pub static ref CONFIG: Config = {
if let Some(path) = &CLI.config {
Config::de(
@ -74,7 +83,9 @@ lazy_static! {
Config::default()
}
};
#[doc(hidden)]
pub static ref CLIENTS: DashMap<String, Client> = DashMap::new();
#[doc(hidden)]
pub static ref RESOLVER: Resolver = {
if CONFIG.stream.dns_servers.is_empty() {
if let Ok((config, opts)) = read_system_conf() {
@ -96,9 +107,11 @@ lazy_static! {
};
}
#[doc(hidden)]
#[global_allocator]
static JEMALLOCATOR: tikv_jemallocator::Jemalloc = tikv_jemallocator::Jemalloc;
#[doc(hidden)]
fn main() -> anyhow::Result<()> {
if CLI.default_config {
println!("{}", Config::default().ser()?);
@ -204,6 +217,7 @@ fn main() -> anyhow::Result<()> {
})
}
#[doc(hidden)]
fn handle_stream(stream: ServerRouteResult, id: String) {
tokio::spawn(async move {
CLIENTS.insert(id.clone(), (DashMap::new(), false));