From a1b92924093e3908a6c5aef1a6215d68d62ae8b1 Mon Sep 17 00:00:00 2001
From: John Doty <john.doty@lacework.net>
Date: Sat, 8 Oct 2022 00:22:22 +0000
Subject: [PATCH] Argument handling

---
 src/main.rs | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/src/main.rs b/src/main.rs
index ab83d6f..7ac268c 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -563,6 +563,11 @@ async fn client_main<Reader: AsyncRead + Unpin, Writer: AsyncWrite + Unpin>(
 #[tokio::main]
 async fn main() {
     let args: Vec<String> = std::env::args().collect();
+    if args.len() != 2 {
+        eprintln!("Usage: fwd <server>");
+        std::process::exit(1);
+    }
+
     let remote = &args[1];
     if remote == "--server" {
         let mut reader = BufReader::new(tokio::io::stdin());