From 3be97adffc2e0bcf63ceec52433e2e8f483d51a4 Mon Sep 17 00:00:00 2001
From: Andrea <andrea@bruschetta.cc>
Date: Mon, 11 Nov 2024 09:25:04 +0100
Subject: [PATCH] Add user input for RSA key pair size in bits

---
 src/main.rs | 13 ++++++++++++-
 1 file changed, 12 insertions(+), 1 deletion(-)

diff --git a/src/main.rs b/src/main.rs
index 98df0f9..bd9a648 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -107,7 +107,18 @@ fn main() {
         Err(error) => println!("error: {error}"),
     }
 
-    let rsa = RSA::new(1024);
+    let mut rsa_size: String = "".to_string();
+
+    info!("Enter the size of the RSA key pair (in bits):");
+    match io::stdin().read_line(&mut rsa_size) {
+        Ok(n) => match n {
+            0 => info!("No input provided"),
+            _ => info!("{} bytes read", n),
+        },
+        Err(error) => println!("error: {error}"),
+    }
+
+    let rsa = RSA::new(rsa_size.trim().parse().unwrap());
 
     info!("Original: {}", message.trim());