hopr_ct_telemetry/

lib.rs

use std::sync::Arc;

use futures::StreamExt;
use hopr_api::{
    PeerId,
    ct::{DestinationRouting, NetworkGraphView, TrafficGeneration},
    network::{NetworkObservations, NetworkView},
};
use hopr_crypto_random::Randomizable;
use hopr_crypto_types::types::OffchainPublicKey;
use hopr_internal_types::protocol::HoprPseudonym;
use hopr_network_types::types::RoutingOptions;
use rand::seq::SliceRandom;
#[cfg(feature = "serde")]
use serde::{Deserialize, Serialize};
use validator::Validate;

/// Configuration for the probing mechanism
#[derive(Debug, Clone, Copy, PartialEq, smart_default::SmartDefault, Validate)]
#[cfg_attr(feature = "serde", derive(Serialize, Deserialize), serde(deny_unknown_fields))]
pub struct ProberConfig {
    /// The delay between individual probing rounds for neighbor discovery
    #[cfg_attr(
        feature = "serde",
        serde(default = "default_probing_interval", with = "humantime_serde")
    )]
    #[default(default_probing_interval())]
    pub interval: std::time::Duration,

    /// The time threshold after which it is reasonable to recheck the nearest neighbor
    #[cfg_attr(
        feature = "serde",
        serde(default = "default_recheck_threshold", with = "humantime_serde")
    )]
    #[default(default_recheck_threshold())]
    pub recheck_threshold: std::time::Duration,
}

/// Delay before repeating probing rounds, must include enough time to traverse NATs
const DEFAULT_REPEATED_PROBING_DELAY: std::time::Duration = std::time::Duration::from_secs(5);

/// Time after which the availability of a node gets rechecked
const DEFAULT_PROBE_RECHECK_THRESHOLD: std::time::Duration = std::time::Duration::from_secs(60);

#[inline]
const fn default_probing_interval() -> std::time::Duration {
    DEFAULT_REPEATED_PROBING_DELAY
}

#[inline]
const fn default_recheck_threshold() -> std::time::Duration {
    DEFAULT_PROBE_RECHECK_THRESHOLD
}

pub struct ImmediateNeighborProber {
    cfg: ProberConfig,
}

impl ImmediateNeighborProber {
    pub fn new(cfg: ProberConfig) -> Self {
        Self { cfg }
    }
}

impl TrafficGeneration for ImmediateNeighborProber {
    fn build<U>(self, network_graph: U) -> impl futures::Stream<Item = DestinationRouting> + Send
    where
        U: NetworkGraphView + Send + Sync + 'static,
    {
        // For each probe target a cached version of transport routing is stored
        let cache_peer_routing: moka::future::Cache<PeerId, DestinationRouting> = moka::future::Cache::builder()
            .time_to_live(std::time::Duration::from_secs(600))
            .max_capacity(100_000)
            .build();

        let cfg = self.cfg;

        futures::stream::repeat(())
            .filter_map(move |_| {
                let nodes = network_graph.nodes();

                async move {
                    hopr_async_runtime::prelude::sleep(cfg.interval).await;
                    Some(nodes)
                }
            })
            .flatten()
            .filter_map(move |peer| {
                let cache_peer_routing = cache_peer_routing.clone();

                async move {
                    cache_peer_routing
                        .try_get_with(peer, async move {
                            Ok::<DestinationRouting, anyhow::Error>(DestinationRouting::Forward {
                                destination: Box::new(OffchainPublicKey::from_peerid(&peer)?.into()),
                                pseudonym: Some(HoprPseudonym::random()),
                                forward_options: RoutingOptions::Hops(0.try_into().expect("0 is a valid u8")),
                                return_options: Some(RoutingOptions::Hops(0.try_into().expect("0 is a valid u8"))),
                            })
                        })
                        .await
                        .ok()
                }
            })
    }
}

#[derive(Clone)]
pub struct ImmediateNeighborChannelGraph<T> {
    network: Arc<T>,
    recheck_threshold: std::time::Duration,
}

impl<T> ImmediateNeighborChannelGraph<T> {
    pub fn new(network: T, recheck_threshold: std::time::Duration) -> Self {
        Self {
            network: Arc::new(network),
            recheck_threshold,
        }
    }
}

#[async_trait::async_trait]
impl<T> hopr_api::ct::NetworkGraphUpdate for ImmediateNeighborChannelGraph<T>
where
    T: NetworkObservations + Send + Sync,
{
    async fn record<N, P>(
        &self,
        telemetry: std::result::Result<hopr_api::ct::Telemetry<N, P>, hopr_api::ct::TrafficGenerationError<P>>,
    ) where
        N: hopr_api::ct::MeasurableNeighbor + Send + Clone,
        P: hopr_api::ct::MeasurablePath + Send + Clone,
    {
        match telemetry {
            Ok(hopr_api::ct::Telemetry::Neighbor(telemetry)) => {
                tracing::trace!(
                    peer = %telemetry.peer(),
                    latency_ms = telemetry.rtt().as_millis(),
                    "neighbor probe successful"
                );
                hopr_api::network::NetworkObservations::update(
                    self.network.as_ref(),
                    telemetry.peer(),
                    Ok(telemetry.rtt() / 2),
                );
            }
            Ok(hopr_api::ct::Telemetry::Loopback(_)) => {
                tracing::warn!(
                    reason = "feature not implemented",
                    "loopback path telemetry not supported yet"
                );
            }
            Err(hopr_api::ct::TrafficGenerationError::ProbeNeighborTimeout(peer)) => {
                hopr_api::network::NetworkObservations::update(self.network.as_ref(), &peer, Err(()));
            }
            Err(hopr_api::ct::TrafficGenerationError::ProbeLoopbackTimeout(_)) => {
                tracing::warn!(
                    reason = "feature not implemented",
                    "loopback path telemetry not supported yet"
                );
            }
        }
    }
}

#[async_trait::async_trait]
impl<T> hopr_api::ct::NetworkGraphView for ImmediateNeighborChannelGraph<T>
where
    T: NetworkView + Send + Sync + Clone + 'static,
{
    fn nodes(&self) -> futures::stream::BoxStream<'static, PeerId> {
        let fetcher = self.network.clone();
        let _recheck_threshold = self.recheck_threshold; // TODO: currently being ignored
        let mut rng = hopr_crypto_random::rng();

        Box::pin(async_stream::stream! {
            let mut peers: Vec<PeerId> = fetcher.discovered_peers().into_iter().collect();
            peers.shuffle(&mut rng);    // shuffle peers to randomize order between rounds

            for peer in peers {
                yield peer;
            }
        })
    }

    async fn find_routes(&self, _destination: &PeerId, _length: usize) -> Vec<DestinationRouting> {
        vec![]
    }
}

#[cfg(test)]
mod tests {
    use std::collections::HashSet;

    use futures::{StreamExt, pin_mut};
    use hopr_api::{
        Multiaddr,
        network::{Health, Observable},
    };
    use hopr_internal_types::NodeId;
    use tokio::time::timeout;

    use super::*;

    const TINY_TIMEOUT: std::time::Duration = std::time::Duration::from_millis(20);

    mockall::mock! {
        Observed {}

        impl Observable for Observed {
            fn record_probe(&mut self, latency: std::result::Result<std::time::Duration, ()>);

            fn last_update(&self) -> std::time::Duration;

            fn average_latency(&self) -> Option<std::time::Duration>;

            fn average_probe_rate(&self) -> f64;

            fn score(&self) -> f64;
        }
    }

    mockall::mock! {
        ScanInteraction {}

        #[async_trait::async_trait]
        impl hopr_api::network::NetworkObservations for ScanInteraction {
            fn update(&self, peer: &PeerId, result: std::result::Result<std::time::Duration, ()>);
        }

        #[async_trait::async_trait]
        impl hopr_api::network::NetworkView for ScanInteraction {
            fn listening_as(&self) -> HashSet<Multiaddr>;

            fn multiaddress_of(&self, peer: &PeerId) -> Option<HashSet<Multiaddr>>;

            fn discovered_peers(&self) -> std::collections::HashSet<PeerId> ;

            fn connected_peers(&self) -> HashSet<PeerId>;

            #[allow(refining_impl_trait)]
            fn observations_for(&self, peer: &PeerId) -> Option<MockObserved>;

            fn health(&self) -> Health;
        }

        impl Clone for ScanInteraction {
            fn clone(&self) -> Self;
        }
    }

    #[tokio::test]
    async fn peers_should_not_be_passed_if_none_are_present() -> anyhow::Result<()> {
        let mut fetcher = MockScanInteraction::new();
        fetcher.expect_discovered_peers().returning(|| HashSet::new());

        let channel_graph = ImmediateNeighborChannelGraph {
            network: Arc::new(fetcher),
            recheck_threshold: ProberConfig::default().recheck_threshold,
        };

        let prober = ImmediateNeighborProber::new(Default::default());
        let stream = prober.build(channel_graph);
        pin_mut!(stream);

        assert!(timeout(TINY_TIMEOUT, stream.next()).await.is_err());

        Ok(())
    }

    lazy_static::lazy_static! {
        static ref RANDOM_PEERS: HashSet<PeerId> = (1..10).map(|_| {
            let peer: PeerId = OffchainPublicKey::from_privkey(&hopr_crypto_random::random_bytes::<32>()).unwrap().into();
            peer
        }).collect::<HashSet<_>>();
    }

    #[tokio::test]
    async fn peers_should_have_randomized_order() -> anyhow::Result<()> {
        let mut fetcher = MockScanInteraction::new();
        fetcher.expect_discovered_peers().returning(|| RANDOM_PEERS.clone());

        let channel_graph = ImmediateNeighborChannelGraph {
            network: Arc::new(fetcher),
            recheck_threshold: ProberConfig::default().recheck_threshold,
        };

        let prober = ImmediateNeighborProber::new(ProberConfig {
            interval: std::time::Duration::from_millis(1),
            ..Default::default()
        });
        let stream = prober.build(channel_graph);
        pin_mut!(stream);

        let actual = timeout(
            TINY_TIMEOUT * 20,
            stream
                .take(RANDOM_PEERS.len())
                .map(|routing| match routing {
                    DestinationRouting::Forward { destination, .. } => {
                        if let NodeId::Offchain(peer_key) = destination.as_ref() {
                            PeerId::from(peer_key)
                        } else {
                            panic!("expected offchain destination, got chain address");
                        }
                    }
                    _ => panic!("expected Forward routing"),
                })
                .collect::<Vec<_>>(),
        )
        .await?;

        assert_eq!(actual.len(), RANDOM_PEERS.len());
        assert!(!actual.iter().zip(RANDOM_PEERS.iter()).all(|(a, b)| a == b));

        Ok(())
    }

    #[tokio::test]
    async fn peers_should_be_generated_in_multiple_rounds_as_long_as_they_are_available() -> anyhow::Result<()> {
        let cfg = ProberConfig {
            interval: std::time::Duration::from_millis(1),
            recheck_threshold: std::time::Duration::from_millis(1000),
            ..Default::default()
        };

        let mut fetcher = MockScanInteraction::new();
        fetcher.expect_discovered_peers().times(2).returning(|| {
            let peer: PeerId = OffchainPublicKey::from_privkey(&hopr_crypto_random::random_bytes::<32>())
                .unwrap()
                .into();
            HashSet::from([peer])
        });
        fetcher.expect_discovered_peers().returning(|| HashSet::new());

        let channel_graph = ImmediateNeighborChannelGraph {
            network: Arc::new(fetcher),
            recheck_threshold: cfg.recheck_threshold,
        };

        let prober = ImmediateNeighborProber::new(cfg);
        let stream = prober.build(channel_graph);
        pin_mut!(stream);

        assert!(timeout(TINY_TIMEOUT, stream.next()).await?.is_some());
        assert!(timeout(TINY_TIMEOUT, stream.next()).await?.is_some());
        assert!(timeout(TINY_TIMEOUT, stream.next()).await.is_err());

        Ok(())
    }
}