move more things from aquatic_http to aquatic_common_tcp

This commit is contained in:
Joakim Frostegård 2020-07-02 16:47:33 +02:00
parent 2e53a2adc1
commit 720596dfb4
6 changed files with 151 additions and 128 deletions

View file

@ -2,26 +2,7 @@ use std::net::SocketAddr;
use serde::{Serialize, Deserialize};
// identical to ws version
#[derive(Debug, Clone, Serialize, Deserialize)]
#[serde(rename_all = "lowercase")]
pub enum LogLevel {
Off,
Error,
Warn,
Info,
Debug,
Trace
}
// identical to ws version
impl Default for LogLevel {
fn default() -> Self {
Self::Error
}
}
pub use aquatic_common_tcp::config::*;
#[derive(Clone, Debug, Serialize, Deserialize)]
@ -46,24 +27,13 @@ pub struct NetworkConfig {
/// Bind to this address
pub address: SocketAddr,
pub ipv6_only: bool,
pub use_tls: bool,
pub tls_pkcs12_path: String,
pub tls_pkcs12_password: String,
#[serde(flatten)]
pub tls: TlsConfig,
pub poll_event_capacity: usize,
pub poll_timeout_milliseconds: u64,
}
// identical to ws version
#[derive(Clone, Debug, Serialize, Deserialize)]
#[serde(default)]
pub struct HandlerConfig {
/// Maximum number of requests to receive from channel before locking
/// mutex and starting work
pub max_requests_per_iter: usize,
pub channel_recv_timeout_microseconds: u64,
}
#[derive(Clone, Debug, Serialize, Deserialize)]
#[serde(default)]
@ -77,31 +47,6 @@ pub struct ProtocolConfig {
}
// identical to ws version
#[derive(Clone, Debug, Serialize, Deserialize)]
#[serde(default)]
pub struct CleaningConfig {
/// Clean peers this often (seconds)
pub interval: u64,
/// Remove peers that haven't announced for this long (seconds)
pub max_peer_age: u64,
/// Remove connections that are older than this (seconds)
pub max_connection_age: u64,
}
// identical to ws version
#[derive(Clone, Debug, Serialize, Deserialize)]
#[serde(default)]
pub struct PrivilegeConfig {
/// Chroot and switch user after binding to sockets
pub drop_privileges: bool,
/// Chroot to this path
pub chroot_path: String,
/// User to switch to after chrooting
pub user: String,
}
impl Default for Config {
fn default() -> Self {
@ -123,9 +68,7 @@ impl Default for NetworkConfig {
Self {
address: SocketAddr::from(([0, 0, 0, 0], 3000)),
ipv6_only: false,
use_tls: false,
tls_pkcs12_path: "".into(),
tls_pkcs12_password: "".into(),
tls: TlsConfig::default(),
poll_event_capacity: 4096,
poll_timeout_milliseconds: 50,
}
@ -141,39 +84,4 @@ impl Default for ProtocolConfig {
peer_announce_interval: 120,
}
}
}
// identical to ws version
impl Default for HandlerConfig {
fn default() -> Self {
Self {
max_requests_per_iter: 10000,
channel_recv_timeout_microseconds: 200,
}
}
}
// identical to ws version
impl Default for CleaningConfig {
fn default() -> Self {
Self {
interval: 30,
max_peer_age: 180,
max_connection_age: 180,
}
}
}
// identical to ws version
impl Default for PrivilegeConfig {
fn default() -> Self {
Self {
drop_privileges: false,
chroot_path: ".".to_string(),
user: "nobody".to_string(),
}
}
}

View file

@ -1,14 +1,13 @@
use std::time::Duration;
use std::fs::File;
use std::io::Read;
use std::sync::Arc;
use std::thread::Builder;
use anyhow::Context;
use native_tls::{Identity, TlsAcceptor};
use parking_lot::Mutex;
use privdrop::PrivDrop;
use aquatic_common_tcp::network::create_tls_acceptor;
pub mod common;
pub mod config;
pub mod handler;
@ -22,7 +21,7 @@ use config::Config;
// almost identical to ws version
pub fn run(config: Config) -> anyhow::Result<()> {
let opt_tls_acceptor = create_tls_acceptor(&config)?;
let opt_tls_acceptor = create_tls_acceptor(&config.network.tls)?;
let state = State::default();
@ -115,29 +114,3 @@ pub fn run(config: Config) -> anyhow::Result<()> {
}
}
// identical to ws version
pub fn create_tls_acceptor(
config: &Config,
) -> anyhow::Result<Option<TlsAcceptor>> {
if config.network.use_tls {
let mut identity_bytes = Vec::new();
let mut file = File::open(&config.network.tls_pkcs12_path)
.context("Couldn't open pkcs12 identity file")?;
file.read_to_end(&mut identity_bytes)
.context("Couldn't read pkcs12 identity file")?;
let identity = Identity::from_pkcs12(
&mut identity_bytes,
&config.network.tls_pkcs12_password
).context("Couldn't parse pkcs12 identity file")?;
let acceptor = TlsAcceptor::new(identity)
.context("Couldn't create TlsAcceptor from pkcs12 identity")?;
Ok(Some(acceptor))
} else {
Ok(None)
}
}