Jędrzej Stuczyński
833114372a
* wip: changes to surb logic + stronger db typing * surb invalidation logic * chore: remove unused deps * resolving todos * a lot of additional bugfixes * 1.88 clippy * wasm fixes * wasm clippy * wallet clippy * wait for epoch end when setting up new network * split ReplyController into Sender and Receiver for easier reasoning * additional reply surbs improvements includes, but is not limited to: unconditionally reseting sender tag on restart, limiting number of surb re-requests, resetting stale surbs on load * fixed calculation of number of removed surbs * add additional calculated field to key rotation info * DBG: 'request_reply_surbs_for_queue_clearing' temp logs * fixes for silly mistakes * conditionally reduce log severity
75 lines
2.6 KiB
Rust
75 lines
2.6 KiB
Rust
// Copyright 2023 - Nym Technologies SA <contact@nymtech.net>
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
use nym_sdk::mixnet::{
|
|
IncludedSurbs, MixnetClient, MixnetMessageSender, Recipient, ReconstructedMessage,
|
|
};
|
|
use nym_service_providers_common::interface::{
|
|
ProviderInterfaceVersion, Request, Response, ResponseContent,
|
|
};
|
|
use nym_socks5_requests::{QueryRequest, Socks5ProtocolVersion, Socks5Request, Socks5Response};
|
|
|
|
fn parse_response(received: Vec<ReconstructedMessage>) -> Socks5Response {
|
|
assert_eq!(received.len(), 1);
|
|
let response: Response<Socks5Request> = Response::try_from_bytes(&received[0].message).unwrap();
|
|
match response.content {
|
|
ResponseContent::Control(control) => panic!("unexpected control response: {control:?}"),
|
|
ResponseContent::ProviderData(data) => data,
|
|
}
|
|
}
|
|
|
|
async fn wait_for_response(client: &mut MixnetClient) -> Socks5Response {
|
|
loop {
|
|
let next = client.wait_for_messages().await.unwrap();
|
|
if !next.is_empty() {
|
|
return parse_response(next);
|
|
}
|
|
}
|
|
}
|
|
|
|
#[tokio::main]
|
|
async fn main() -> anyhow::Result<()> {
|
|
//nym_bin_common::logging::setup_logging();
|
|
let mut client = MixnetClient::connect_new().await.unwrap();
|
|
let provider: Recipient = "AN8eLxYWFitCkMn92zim3PrPszxJZDYyFFKP7qnnAAew.8UAxL3LwQBis6WpM3GGXaqKGaVdnLCpGJWumHT6KNdTH@77TSuVU8d1oXKbPzjec2xh4i3Wj5WwUyy9Lr36sm8gZm".parse().unwrap();
|
|
|
|
let open_proxy_request = Request::new_provider_data(
|
|
ProviderInterfaceVersion::new_current(),
|
|
Socks5Request::new_query(
|
|
Socks5ProtocolVersion::new_current(),
|
|
QueryRequest::OpenProxy,
|
|
),
|
|
);
|
|
let description_request = Request::new_provider_data(
|
|
ProviderInterfaceVersion::new_current(),
|
|
Socks5Request::new_query(
|
|
Socks5ProtocolVersion::new_current(),
|
|
QueryRequest::Description,
|
|
),
|
|
);
|
|
println!("Sending 'OpenProxy' query...");
|
|
client
|
|
.send_message(
|
|
provider,
|
|
open_proxy_request.into_bytes(),
|
|
IncludedSurbs::new(10),
|
|
)
|
|
.await?;
|
|
let response = wait_for_response(&mut client).await;
|
|
println!("response to 'OpenProxy' query: {response:#?}");
|
|
|
|
println!("Sending 'Description' query...");
|
|
client
|
|
.send_message(
|
|
provider,
|
|
description_request.into_bytes(),
|
|
IncludedSurbs::none(),
|
|
)
|
|
.await?;
|
|
let response = wait_for_response(&mut client).await;
|
|
println!("response to 'Description' query: {response:#?}");
|
|
|
|
client.disconnect().await;
|
|
Ok(())
|
|
}
|