aquatic_http access list: add config, state field, initial load

This commit is contained in:
Joakim Frostegård 2021-10-15 22:38:58 +02:00
parent b8b9a9839a
commit 4fa199a1e0
3 changed files with 18 additions and 0 deletions

View file

@ -1,6 +1,7 @@
use std::net::{Ipv4Addr, Ipv6Addr, SocketAddr}; use std::net::{Ipv4Addr, Ipv6Addr, SocketAddr};
use std::sync::Arc; use std::sync::Arc;
use aquatic_common::access_list::AccessList;
use crossbeam_channel::{Receiver, Sender}; use crossbeam_channel::{Receiver, Sender};
use either::Either; use either::Either;
use hashbrown::HashMap; use hashbrown::HashMap;
@ -116,12 +117,14 @@ pub struct TorrentMaps {
#[derive(Clone)] #[derive(Clone)]
pub struct State { pub struct State {
pub torrent_maps: Arc<Mutex<TorrentMaps>>, pub torrent_maps: Arc<Mutex<TorrentMaps>>,
pub access_list: Arc<Mutex<AccessList>>,
} }
impl Default for State { impl Default for State {
fn default() -> Self { fn default() -> Self {
Self { Self {
torrent_maps: Arc::new(Mutex::new(TorrentMaps::default())), torrent_maps: Arc::new(Mutex::new(TorrentMaps::default())),
access_list: Arc::new(Mutex::new(AccessList::default())),
} }
} }
} }

View file

@ -1,5 +1,6 @@
use std::net::SocketAddr; use std::net::SocketAddr;
use aquatic_common::access_list::AccessListConfig;
use serde::{Deserialize, Serialize}; use serde::{Deserialize, Serialize};
use aquatic_cli_helpers::LogLevel; use aquatic_cli_helpers::LogLevel;
@ -21,6 +22,7 @@ pub struct Config {
pub cleaning: CleaningConfig, pub cleaning: CleaningConfig,
pub statistics: StatisticsConfig, pub statistics: StatisticsConfig,
pub privileges: PrivilegeConfig, pub privileges: PrivilegeConfig,
pub access_list: AccessListConfig,
} }
impl aquatic_cli_helpers::Config for Config { impl aquatic_cli_helpers::Config for Config {
@ -111,6 +113,7 @@ impl Default for Config {
cleaning: CleaningConfig::default(), cleaning: CleaningConfig::default(),
statistics: StatisticsConfig::default(), statistics: StatisticsConfig::default(),
privileges: PrivilegeConfig::default(), privileges: PrivilegeConfig::default(),
access_list: AccessListConfig::default(),
} }
} }
} }

View file

@ -7,6 +7,8 @@ use mio::{Poll, Waker};
use parking_lot::Mutex; use parking_lot::Mutex;
use privdrop::PrivDrop; use privdrop::PrivDrop;
use aquatic_common::access_list::AccessListMode;
pub mod common; pub mod common;
pub mod config; pub mod config;
pub mod handler; pub mod handler;
@ -22,6 +24,16 @@ pub const APP_NAME: &str = "aquatic_http: HTTP/TLS BitTorrent tracker";
pub fn run(config: Config) -> anyhow::Result<()> { pub fn run(config: Config) -> anyhow::Result<()> {
let state = State::default(); let state = State::default();
match config.access_list.mode {
AccessListMode::Require | AccessListMode::Forbid => {
state
.access_list
.lock()
.update_from_path(&config.access_list.path)?;
}
AccessListMode::Ignore => {}
}
start_workers(config.clone(), state.clone())?; start_workers(config.clone(), state.clone())?;
loop { loop {