From b2a73e561241c45f870df84519e6beb027f36de0 Mon Sep 17 00:00:00 2001
From: daleclack <daleclack1@foxmail.com>
Date: Fri, 18 Feb 2022 18:57:17 +0800
Subject: [PATCH] Update XeRelease Rust Edition

---
 Rust_tests/XeRelease_Rust/src/main.rs | 15 ++++++++++-----
 1 file changed, 10 insertions(+), 5 deletions(-)

diff --git a/Rust_tests/XeRelease_Rust/src/main.rs b/Rust_tests/XeRelease_Rust/src/main.rs
index 2a53c19..d65cc5e 100644
--- a/Rust_tests/XeRelease_Rust/src/main.rs
+++ b/Rust_tests/XeRelease_Rust/src/main.rs
@@ -12,14 +12,19 @@ fn main() {
     // Input mode selection
     let mut string1 = String::new();
     io::stdin().read_line(&mut string1).expect("Read Error!");
-    if !string1.trim().is_empty(){
-        // Get index for mode
-        let index: usize = string1.trim().parse().unwrap();
+    let index_string = string1.trim();
+    if !index_string.is_empty()
+        && (index_string == "0"
+            || index_string == "1"
+            || index_string == "2"
+            || index_string == "3")
+    {
+        // Get index for mode5
+        let index: usize = index_string.parse::<usize>().unwrap();
         funcs[index]();
-    }else{
+    } else {
         println!("Please input a vaild mode index!");
     }
-    
 }
 
 fn about() {