Jon Häggblad
46e2c74a98
* Rename to nym-task * Update imports to use nym_task * Update to nym_task in explicit module prefix * Touchups and rustfmt * rustfmt * crate metadata * fix nym-connect-android
78 lines
2.6 KiB
Rust
78 lines
2.6 KiB
Rust
// Copyright 2021 - Nym Technologies SA <contact@nymtech.net>
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
use crate::client::{Client, Config, SendWithoutResponse};
|
|
use futures::channel::mpsc;
|
|
use futures::StreamExt;
|
|
use log::*;
|
|
use nym_sphinx::forwarding::packet::MixPacket;
|
|
use std::time::Duration;
|
|
|
|
pub type MixForwardingSender = mpsc::UnboundedSender<MixPacket>;
|
|
type MixForwardingReceiver = mpsc::UnboundedReceiver<MixPacket>;
|
|
|
|
/// A specialisation of client such that it forwards any received packets on the channel into the
|
|
/// mix network immediately, i.e. will not try to listen for any responses.
|
|
pub struct PacketForwarder {
|
|
mixnet_client: Client,
|
|
packet_receiver: MixForwardingReceiver,
|
|
shutdown: nym_task::TaskClient,
|
|
}
|
|
|
|
impl PacketForwarder {
|
|
pub fn new(
|
|
initial_reconnection_backoff: Duration,
|
|
maximum_reconnection_backoff: Duration,
|
|
initial_connection_timeout: Duration,
|
|
maximum_connection_buffer_size: usize,
|
|
use_legacy_version: bool,
|
|
shutdown: nym_task::TaskClient,
|
|
) -> (PacketForwarder, MixForwardingSender) {
|
|
let client_config = Config::new(
|
|
initial_reconnection_backoff,
|
|
maximum_reconnection_backoff,
|
|
initial_connection_timeout,
|
|
maximum_connection_buffer_size,
|
|
use_legacy_version,
|
|
);
|
|
|
|
let (packet_sender, packet_receiver) = mpsc::unbounded();
|
|
|
|
(
|
|
PacketForwarder {
|
|
mixnet_client: Client::new(client_config),
|
|
packet_receiver,
|
|
shutdown,
|
|
},
|
|
packet_sender,
|
|
)
|
|
}
|
|
|
|
pub async fn run(&mut self) {
|
|
while !self.shutdown.is_shutdown() {
|
|
tokio::select! {
|
|
biased;
|
|
_ = self.shutdown.recv() => {
|
|
log::trace!("PacketForwarder: Received shutdown");
|
|
}
|
|
Some(mix_packet) = self.packet_receiver.next() => {
|
|
trace!("Going to forward packet to {:?}", mix_packet.next_hop());
|
|
|
|
let next_hop = mix_packet.next_hop();
|
|
let packet_mode = mix_packet.packet_mode();
|
|
let sphinx_packet = mix_packet.into_sphinx_packet();
|
|
// we don't care about responses, we just want to fire packets
|
|
// as quickly as possible
|
|
|
|
if let Err(err) =
|
|
self.mixnet_client
|
|
.send_without_response(next_hop, sphinx_packet, packet_mode)
|
|
{
|
|
debug!("failed to forward the packet - {err}")
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|