void/ZeroTierOne
2
0
Fork 0
mirror of https://github.com/zerotier/ZeroTierOne.git synced 2025-04-26 08:57:26 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

cleanup

Browse source
This commit is contained in:
Adam Ierymenko 2022-10-21 17:36:34 -07:00
parent c7b6a997be
commit 7cb52b112a
No known key found for this signature in database
GPG key ID: C8877CF2D7A5D7F3
1 changed files with 11 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

22
network-hypervisor/src/vl1/path.rs
View file

@ -5,7 +5,7 @@ use std::hash::{BuildHasher, Hasher};
use std::sync::atomic::{AtomicI64, Ordering};
use std::sync::Mutex;
use crate::protocol::*;
use crate::protocol;
use crate::vl1::endpoint::Endpoint;
use crate::vl1::node::*;
@ -13,7 +13,7 @@ use zerotier_crypto::random;
use zerotier_utils::pocket::Pocket;
use zerotier_utils::NEVER_HAPPENED_TICKS;
pub(crate) const SERVICE_INTERVAL_MS: i64 = PATH_KEEPALIVE_INTERVAL;
pub(crate) const SERVICE_INTERVAL_MS: i64 = protocol::PATH_KEEPALIVE_INTERVAL;
pub(crate) enum PathServiceResult {
Ok,
@ -32,7 +32,7 @@ pub struct Path {
last_send_time_ticks: AtomicI64,
last_receive_time_ticks: AtomicI64,
create_time_ticks: i64,
fragmented_packets: Mutex<HashMap<u64, v1::FragmentedPacket, PacketIdHasher>>,
fragmented_packets: Mutex<HashMap<u64, protocol::v1::FragmentedPacket, PacketIdHasher>>,
}
impl Path {
@ -70,15 +70,15 @@ impl Path {
packet_id: u64,
fragment_no: u8,
fragment_expecting_count: u8,
packet: PooledPacketBuffer,
packet: protocol::PooledPacketBuffer,
time_ticks: i64,
) -> Option<v1::FragmentedPacket> {
) -> Option<protocol::v1::FragmentedPacket> {
let mut fp = self.fragmented_packets.lock().unwrap();
// Discard some old waiting packets if the total incoming fragments for a path exceeds a
// sanity limit. This is to prevent memory exhaustion DOS attacks.
let fps = fp.len();
if fps > v1::FRAGMENT_MAX_INBOUND_PACKETS_PER_PATH {
if fps > protocol::v1::FRAGMENT_MAX_INBOUND_PACKETS_PER_PATH {
let mut entries: Vec<(i64, u64)> = Vec::new();
entries.reserve(fps);
for f in fp.iter() {
@ -92,7 +92,7 @@ impl Path {
if fp
.entry(packet_id)
.or_insert_with(|| v1::FragmentedPacket::new(time_ticks))
.or_insert_with(|| protocol::v1::FragmentedPacket::new(time_ticks))
.add_fragment(packet, fragment_no, fragment_expecting_count)
{
fp.remove(&packet_id)
@ -115,15 +115,15 @@ impl Path {
self.fragmented_packets
.lock()
.unwrap()
.retain(|_, frag| (time_ticks - frag.ts_ticks) < v1::FRAGMENT_EXPIRATION);
if (time_ticks - self.last_receive_time_ticks.load(Ordering::Relaxed)) < PATH_EXPIRATION_TIME {
if (time_ticks - self.last_send_time_ticks.load(Ordering::Relaxed)) >= PATH_KEEPALIVE_INTERVAL {
.retain(|_, frag| (time_ticks - frag.ts_ticks) < protocol::v1::FRAGMENT_EXPIRATION);
if (time_ticks - self.last_receive_time_ticks.load(Ordering::Relaxed)) < protocol::PATH_EXPIRATION_TIME {
if (time_ticks - self.last_send_time_ticks.load(Ordering::Relaxed)) >= protocol::PATH_KEEPALIVE_INTERVAL {
self.last_send_time_ticks.store(time_ticks, Ordering::Relaxed);
PathServiceResult::NeedsKeepalive
} else {
PathServiceResult::Ok
}
} else if (time_ticks - self.create_time_ticks) < PATH_EXPIRATION_TIME {
} else if (time_ticks - self.create_time_ticks) < protocol::PATH_EXPIRATION_TIME {
PathServiceResult::Ok
} else {
PathServiceResult::Dead