summaryrefslogtreecommitdiff
path: root/lab5/src/check_window.rs
blob: 216f1dcb83a9893fa01c3d74f754f0e86ea8a8c2 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
use crate::utils::{adler64, powmod};
use crate::window_state::WindowState;
use eframe::egui;

pub struct Window {
    message: String,
    signature: String,
    key: String,
    correct: Option<bool>,
    error: Option<String>,
}

impl Default for Window {
    fn default() -> Self {
        Self {
            message: String::new(),
            signature: String::new(),
            key: String::new(),
            correct: None,
            error: None,
        }
    }
}

impl Window {
    fn check(&mut self) {
        let split = self.key.split(",").collect::<Vec<&str>>();
        if split.len() != 2 {
            self.correct = None;
            self.error = Some("Неверный формат ключа".to_string());
            return;
        }
        let y = match split[0].parse::<u64>() {
            Ok(num) => num,
            Err(_) => {
                self.correct = None;
                self.error = Some("Не удалось прочитать открытый ключ".to_string());
                return;
            }
        };
        let p = match split[1].parse::<u64>() {
            Ok(num) => num,
            Err(_) => {
                self.correct = None;
                self.error = Some("Не удалось прочитать модуль".to_string());
                return;
            }
        };

        let signature = match self.signature.parse::<u64>() {
            Ok(num) => num,
            Err(_) => {
                self.correct = None;
                self.error = Some("Не удалось прочитать подпись".to_string());
                return;
            }
        };

        let h = adler64(self.message.as_bytes()) % p;
        self.correct = Some(powmod(signature, h, p) == y);
    }
}

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.message));
        ui.label("Подпись:");
        let resp2 = ui.add(egui::TextEdit::singleline(&mut self.signature));
        ui.label("Открытый ключ:");
        let resp3 = ui.add(egui::TextEdit::singleline(&mut self.key));
        if resp1.changed() || resp2.changed() || resp3.changed() {
            self.correct = None;
        }

        if ui.button("Проверить подпись").clicked() {
            self.check();
        }

        if let Some(true) = &self.correct {
            ui.label("Подпись верна");
        } else if let Some(false) = &self.correct {
            ui.label("Подпись не соответствует сообщению");
        }
    }
}