1 files changed, 114 insertions, 0 deletions
diff --git a/lab2/src/decrypt_window.rs b/lab2/src/decrypt_window.rs
new file mode 100644
index 0000000..aee9893
--- /dev/null
+++ b/lab2/src/decrypt_window.rs
@@ -0,0 +1,114 @@
+use crate::utils::{from_bytes, powmod};
+use crate::window_state::WindowState;
+use base64;
+use eframe::egui;
+
+pub struct Window {
+    ciphertext: String,
+    key: String,
+    message: Option<String>,
+    error: Option<String>,
+}
+
+impl Default for Window {
+    fn default() -> Self {
+        Self {
+            ciphertext: String::new(),
+            key: String::new(),
+            message: None,
+            error: None,
+        }
+    }
+}
+
+impl Window {
+    fn decrypt(&mut self) {
+        let split = self.key.split(",").collect::<Vec<&str>>();
+        if split.len() != 2 {
+            self.message = None;
+            self.error = Some("Неверный формат ключа".to_string());
+            return;
+        }
+        let d = match split[0].parse::<u64>() {
+            Ok(num) => num,
+            Err(_) => {
+                self.message = None;
+                self.error = Some("Не удалось прочитать открытую экспоненту".to_string());
+                return;
+            }
+        };
+        let n = match split[1].parse::<u64>() {
+            Ok(num) => num,
+            Err(_) => {
+                self.message = None;
+                self.error = Some("Не удалось прочитать модуль".to_string());
+                return;
+            }
+        };
+
+        let cipher = match base64::decode(&self.ciphertext) {
+            Ok(v) => v,
+            Err(_) => {
+                self.message = None;
+                self.error = Some("Введён некорректный шифротекст".to_string());
+                return;
+            }
+        };
+        if cipher.len() == 0 || cipher.len() % 4 != 0 {
+            self.message = None;
+            self.error = Some("Введён некорректный шифротекст".to_string());
+            return;
+        }
+        let mut nums = Vec::new();
+        for i in 0..cipher.len() / 4 {
+            let num = from_bytes(
+                cipher[4 * i],
+                cipher[4 * i + 1],
+                cipher[4 * i + 2],
+                cipher[4 * i + 3],
+            );
+            nums.push(powmod(num, d, n) as u8);
+        }
+        let message = match String::from_utf8(nums) {
+            Ok(m) => m,
+            Err(_) => {
+                self.message = None;
+                self.error = Some("Расшифрованное сообщение не является UTF-8".to_string());
+                return;
+            }
+        };
+        self.message = Some(message);
+    }
+}
+
+impl WindowState for Window {
+    fn get_name(&self) -> &str {
+        "Расшифрование сообщения"
+    }
+
+    fn update(&mut self, ui: &mut egui::Ui) {
+        ui.label("Шифротекст:");
+        let resp1 = ui.add(egui::TextEdit::multiline(&mut self.ciphertext));
+        ui.label("Закрытый ключ:");
+        let resp2 = ui.add(egui::TextEdit::singleline(&mut self.key));
+        if resp1.changed() || resp2.changed() {
+            self.message = None;
+            self.error = None;
+        }
+        if ui.button("Расшифровать").clicked() {
+            self.decrypt();
+        }
+
+        if let Some(text) = &self.message {
+            let mut tmp = text.clone();
+            ui.label("Расшифрованное сообщение:");
+            ui.add_enabled(false, egui::TextEdit::multiline(&mut tmp));
+            if ui.button("Скопировать сообщение").clicked() {
+                ui.output().copied_text = tmp.clone();
+            }
+        }
+        if let Some(error) = &self.error {
+            ui.label(error);
+        }
+    }
+}