server-security 0.1.1

server security proxy
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

use crate::proxy::ValidateType;
use once_cell::sync::Lazy;
use std::collections::HashMap;
use std::net::SocketAddr;
use std::sync::RwLock;

static IP_TABLE: Lazy<RwLock<HashMap<String, ()>>> = Lazy::new(Default::default);

pub fn validate(remote_addr: &SocketAddr) -> anyhow::Result<ValidateType> {
    let remote_ip = remote_addr.ip().to_string();
    let mut guard = IP_TABLE.write().unwrap();

    // if guard.len() > 0 {
    //     TODO change configuration
    // return Err(anyhow::anyhow!("{} not exit error", remote_ip));
    // }

    if guard.get(&remote_ip).is_some() {
        return Ok(ValidateType::Normal);
    }

    guard.insert(remote_ip, ());

    Ok(ValidateType::Normal)
}