TLS transport and games framework on client and server

1 files changed, 10 insertions, 1 deletions

diff --git a/src/bin/server/args.rs b/src/bin/server/args.rs
index 31c0acc..fc844ea 100644
--- a/src/bin/server/args.rs
+++ b/src/bin/server/args.rs
@@ -1,12 +1,14 @@
 pub struct Args {
     pub ip: String,
     pub port: u16,
+    pub unsafe_mode: bool,
 }
 
 pub fn parse_args() -> Args {
     let args: Vec<String> = std::env::args().collect();
     let mut ip = "127.0.0.1".to_string();
     let mut port = 11525;
+    let mut unsafe_mode = false;
     let mut i = 1;
     while i < args.len() {
         if args[i] == "-p" {
@@ -32,10 +34,17 @@ pub fn parse_args() -> Args {
                 eprintln!("-i requires an IP address");
                 std::process::exit(1);
             }
+        } else if args[i] == "--unsafe" {
+            unsafe_mode = true;
+            i += 1;
         } else {
             eprintln!("Unknown argument: {}", args[i]);
             std::process::exit(1);
         }
     }
-    Args { ip, port }
+    Args {
+        ip,
+        port,
+        unsafe_mode,
+    }
 }