use clap::Parser;
use tracing_subscriber::filter::LevelFilter;
#[derive(clap::Parser)]
#[command(version, about)]
struct Args {
#[arg(short = 'c', long, default_value = "tsrs_keys.json")]
key_file: std::path::PathBuf,
#[arg(short = 'k', long)]
auth_key: Option<String>,
#[clap(short, long, default_value_t = 1234)]
listen_port: u16,
}
#[tokio::main(flavor = "multi_thread")]
async fn main() -> Result<(), Box<dyn core::error::Error>> {
tracing_subscriber::fmt()
.with_env_filter(
tracing_subscriber::EnvFilter::builder()
.with_default_directive(LevelFilter::INFO.into())
.from_env_lossy(),
)
.init();
let args = Args::parse();
let dev = tailscale::Device::new(
&tailscale::Config {
key_state: tailscale::load_key_file(&args.key_file, Default::default()).await?,
..Default::default()
},
args.auth_key,
)
.await?;
let sockaddr = (dev.ipv4_addr().await?, args.listen_port).into();
let listener = dev.tcp_listen(sockaddr).await?;
tracing::info!(listening_addr = %sockaddr);
loop {
let conn = listener.accept().await?;
tokio::task::spawn(async move {
let remote_ep = conn.remote_addr();
tracing::info!(%remote_ep, "accepted connection");
let (mut reader, mut writer) = tokio::io::split(conn);
if let Err(e) = tokio::io::copy(&mut reader, &mut writer).await {
tracing::error!(%remote_ep, error = %e);
} else {
tracing::info!(%remote_ep, "remote hung up");
}
});
}
}