72 lines
1.6 KiB
Rust
72 lines
1.6 KiB
Rust
use std::env;
|
|
|
|
fn help() {
|
|
println!(
|
|
"Usage: sclient [--servername <string>] [-k | --insecure] [ssh|rsync] <remote> <port>
|
|
ex: sclient whatever.com 3000
|
|
(whatever.com:443 localhost:3000)
|
|
ex: sclient whatever.com:4080 0.0.0.0:3000"
|
|
);
|
|
}
|
|
|
|
fn main() {
|
|
let args: Vec<String> = env::args().collect();
|
|
|
|
if args
|
|
.iter()
|
|
.any(|element| element == "-h" || element == "--help")
|
|
{
|
|
help();
|
|
return;
|
|
}
|
|
|
|
let mut flags = Flags::default();
|
|
|
|
// try!()
|
|
if args
|
|
.iter()
|
|
.any(|element| element == "-k" || element == "--insecure")
|
|
{
|
|
flags.insecure = true;
|
|
}
|
|
|
|
let serverNameArg = args
|
|
.iter()
|
|
.enumerate()
|
|
.find(|(index, element)| *element == "-servername");
|
|
|
|
if let Some((index, _)) = serverNameArg {
|
|
flags.servername = args[index + 1].clone();
|
|
}
|
|
|
|
// get port
|
|
let maybe_port = args[args.len() - 1].parse::<i32>();
|
|
if maybe_port.is_ok() {
|
|
// pull out of maybe option error
|
|
flags.port = maybe_port.unwrap();
|
|
} else {
|
|
help();
|
|
print!("<port> needs to be an integer");
|
|
}
|
|
// get remote
|
|
flags.remote = args[args.len() - 2].clone();
|
|
|
|
if args[args.len() - 3] == "ssh" || args[args.len() - 3] == "rsync" {
|
|
flags.custom = args[args.len() - 3].clone();
|
|
}
|
|
|
|
dbg!(flags);
|
|
// tls port forward functionality
|
|
// https://github.com/izderadicka/ptunnel-rust
|
|
// https://github.com/ctz/rustls
|
|
}
|
|
|
|
#[derive(Default, Debug)]
|
|
struct Flags {
|
|
insecure: bool,
|
|
servername: String,
|
|
remote: String,
|
|
custom: String,
|
|
port: i32,
|
|
}
|