mirror of
https://github.com/EasyTier/EasyTier.git
synced 2026-05-15 18:35:47 +00:00
fix packet split on udp tunnel and avoid tcp proxy access rpc portal (#2107)
* distinct control / data when forward packets * fix rpc split for udp tunnel * feat(easytier-web): pass public ip in validate token webhook * protect rpc port from subnet proxy
This commit is contained in:
@@ -0,0 +1,61 @@
|
||||
use std::collections::HashMap;
|
||||
use std::sync::Mutex;
|
||||
|
||||
use once_cell::sync::Lazy;
|
||||
|
||||
static PROTECTED_TCP_PORTS: Lazy<Mutex<HashMap<u16, usize>>> =
|
||||
Lazy::new(|| Mutex::new(HashMap::new()));
|
||||
|
||||
pub fn register_protected_tcp_port(port: u16) {
|
||||
let mut ports = PROTECTED_TCP_PORTS.lock().unwrap();
|
||||
*ports.entry(port).or_default() += 1;
|
||||
}
|
||||
|
||||
pub fn unregister_protected_tcp_port(port: u16) {
|
||||
let mut ports = PROTECTED_TCP_PORTS.lock().unwrap();
|
||||
if let Some(ref_count) = ports.get_mut(&port) {
|
||||
*ref_count -= 1;
|
||||
if *ref_count == 0 {
|
||||
ports.remove(&port);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
pub fn is_protected_tcp_port(port: u16) -> bool {
|
||||
PROTECTED_TCP_PORTS.lock().unwrap().contains_key(&port)
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
pub fn clear_protected_tcp_ports_for_test() {
|
||||
PROTECTED_TCP_PORTS.lock().unwrap().clear();
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::{
|
||||
clear_protected_tcp_ports_for_test, is_protected_tcp_port, register_protected_tcp_port,
|
||||
unregister_protected_tcp_port,
|
||||
};
|
||||
|
||||
#[test]
|
||||
fn protected_tcp_port_registry_is_ref_counted() {
|
||||
clear_protected_tcp_ports_for_test();
|
||||
|
||||
register_protected_tcp_port(15888);
|
||||
register_protected_tcp_port(15888);
|
||||
assert!(is_protected_tcp_port(15888));
|
||||
|
||||
unregister_protected_tcp_port(15888);
|
||||
assert!(is_protected_tcp_port(15888));
|
||||
|
||||
unregister_protected_tcp_port(15888);
|
||||
assert!(!is_protected_tcp_port(15888));
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn unregistering_unknown_port_is_a_noop() {
|
||||
clear_protected_tcp_ports_for_test();
|
||||
unregister_protected_tcp_port(15888);
|
||||
assert!(!is_protected_tcp_port(15888));
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user