quicnet 0.2.2

minimal peer-to-peer network protocol over QUIC
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42

// access control list for incoming connections
use crate::PeerId;
use anyhow::Result;
use std::collections::HashSet;
use std::fs;
use std::path::Path;

pub struct AuthorizedPeers {
    peers: Option<HashSet<PeerId>>, // None = allow all
}

impl AuthorizedPeers {
    pub fn load() -> Result<Self> {
        let path = dirs::home_dir()
            .ok_or_else(|| anyhow::anyhow!("no home dir"))?
            .join(".quicnet/authorized_peers");
        Self::load_path(&path)
    }

    pub fn load_path<P: AsRef<Path>>(path: P) -> Result<Self> {
        let path = path.as_ref();
        if !path.exists() {
            return Ok(Self { peers: None }); // allow all if no file
        }
        let mut peers = HashSet::new();
        for line in fs::read_to_string(path)?.lines() {
            let line = line.trim();
            if !line.is_empty() && !line.starts_with('#')
                && let Ok(peer_id) = line.parse::<PeerId>() {
                    peers.insert(peer_id);
                }
        }
        Ok(Self { peers: Some(peers) })
    }

    pub fn is_authorized(&self, peer_id: &PeerId) -> bool {
        match &self.peers {
            None => true,
            Some(list) => list.contains(peer_id),
        }
    }
}