Jędrzej Stuczyński
114d92f93f
* changed NymConfigTemplate trait to call 'template' by reference * network requester lib * introduced generic parameter to 'MixTrafficController' to allow non-remote gateways * allowing for custom gateway sender * types cleanup * minor gateway cmds refactor + initial NR work * wip * running a NR inside gateway note: this NR isnt tied to the gateway yet * rebase fixes * propagating same shutdown handle * wip * starting NR with appropriate local transceiver * fixed premature shutdown * wiring up PacketRouter * both ends wired together * actually working so much cleanup to do now * started removing dead code * wip * temp: hardcode gateway * further cleanup * fixed build of other binaries * setup-network-requester subcmd * overriding NR config in gateway init/run * wip making it wasm-compatible [again] * refactored 'GatewaySetup' * clippy and friends * removed debug code * rust 1.72 lints * ensuring local gateway is available + some comments * correctly putting network requester data in the same underlying details struct * improved gateway errors * changed 'network_requester_config' deserialization * missing clap annotation for 'enabled' flag in 'setup-network-requester' command * saving config file after 'setup-network-requester' * removed dead code * review comments * make embedded NR wait for gateway to come online (for at most 70min) * fixed shutdown on successful gateway wait * updated NR config override
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_type = mix_packet.packet_type();
|
|
let packet = mix_packet.into_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, packet, packet_type)
|
|
{
|
|
debug!("failed to forward the packet - {err}")
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|