YGGverse/aquatic
1
0
Fork 0
mirror of https://github.com/YGGverse/aquatic.git synced 2026-03-31 17:55:36 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

aquatic_udp: mio: access list: use SIGUSR1, improve errors

Browse source
This commit is contained in:
Joakim Frostegård 2021-11-02 21:06:57 +01:00
parent 78d29770f3
commit 9b75b50802
2 changed files with 35 additions and 29 deletions
Download patch file Download diff file Expand all files Collapse all files

31
aquatic_common/src/access_list.rs
View file

@ -3,6 +3,7 @@ use std::io::{BufRead, BufReader};
use std::path::PathBuf;
use std::sync::Arc;
use anyhow::Context;
use arc_swap::ArcSwap;
use hashbrown::HashSet;
use serde::{Deserialize, Serialize};
@ -59,7 +60,11 @@ impl AccessList {
let mut new_list = Self::default();
for line in reader.lines() {
new_list.insert_from_line(&line?)?;
let line = line?;
new_list
.insert_from_line(&line)
.with_context(|| format!("Invalid line in access list: {}", line))?;
}
Ok(new_list)
@ -75,31 +80,17 @@ impl AccessList {
}
pub trait AccessListQuery {
fn update(&self, config: &AccessListConfig);
fn update(&self, config: &AccessListConfig) -> anyhow::Result<()>;
fn allows(&self, list_mode: AccessListMode, info_hash_bytes: &[u8; 20]) -> bool;
}
pub type AccessListArcSwap = ArcSwap<AccessList>;
impl AccessListQuery for AccessListArcSwap {
fn update(&self, config: &AccessListConfig) {
match config.mode {
AccessListMode::White | AccessListMode::Black => {
match AccessList::create_from_path(&config.path) {
Ok(new) => {
self.store(Arc::new(new));
}
Err(err) => {
::log::error!("Updating access list failed: {:?}", err);
}
}
}
AccessListMode::Off => {
::log::error!(
"AccessListQuery::update_from_path called, but AccessListMode is Off"
);
}
}
fn update(&self, config: &AccessListConfig) -> anyhow::Result<()> {
self.store(Arc::new(AccessList::create_from_path(&config.path)?));
Ok(())
}
fn allows(&self, mode: AccessListMode, info_hash_bytes: &[u8; 20]) -> bool {