Jon Häggblad
0fa1961a04
* Initial version * Add open_proxy request * Start adding cli commands * use commands * Common response type * json output * Tidy * Remove the All case * Remove empty file * Remove todo * Add ping command * Ping 4 times by default * Logic for controlling number of loops * Some print tweaks
73 lines
2.5 KiB
Rust
73 lines
2.5 KiB
Rust
// Copyright 2023 - Nym Technologies SA <contact@nymtech.net>
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
use nym_sdk::mixnet::{IncludedSurbs, MixnetClient, 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_bytes(
|
|
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_bytes(
|
|
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(())
|
|
}
|