megalock 0.1.0

X11 xlock replacement
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
91
92
93
94
95
96
97
98

mod pam;
pub mod statics;

use crate::{utils::get_username, wm::pam::authenticate_password};
use anyhow::Result;
use std::sync::{Arc, Mutex};
use tracing::debug;

#[derive(Debug, Clone)]
pub enum Call {
    SelectFocusedWindow,
    OpenFullscreenWindow(String),
    GrabPointerAndKeyboard,
    UnlockSleep,
    RaiseWindow,
    RedrawScreen,
    FlushCommands,
}

pub trait Broker {
    fn set_pam_return(&mut self, pam: Arc<Mutex<Option<std::sync::mpsc::Receiver<()>>>>);
    fn pam_return(&self) -> Arc<Mutex<Option<std::sync::mpsc::Receiver<()>>>>;
    fn set_pam(&mut self, pam: std::sync::mpsc::Sender<()>);
    fn pam(&self) -> Option<std::sync::mpsc::Sender<()>>;
    fn set_receiver(&mut self, call: Arc<Mutex<Option<std::sync::mpsc::Receiver<Call>>>>);
    fn receiver(&self) -> Arc<Mutex<Option<std::sync::mpsc::Receiver<Call>>>>;
    fn listen(&mut self) -> Result<()>;
}

pub trait Client {
    fn pam_return(&self) -> Arc<Mutex<std::sync::mpsc::Sender<()>>>;
    fn pam(&self) -> Arc<Mutex<std::sync::mpsc::Receiver<()>>>;
    fn call(&self) -> Option<std::sync::mpsc::Sender<Call>>;
    fn clear_password(&mut self);
    fn password(&self) -> String;

    fn loop_until_pam(&mut self, name: &str) -> Result<()> {
        while let Ok(_) = self.pam().lock().unwrap().recv() {
            debug!(
                "PAM authentication attempt: username: '{}', password: '{}'",
                get_username()?,
                self.password()
            );

            match authenticate_password(name) {
                Ok(_) => {
                    debug!("PAM authentication successful");
                    return Ok(());
                }
                Err(e) => {
                    debug!("PAM authentication failed: {}", e);
                    self.clear_password();
                }
            }

            self.pam_return().lock().unwrap().send(()).unwrap();
        }

        Ok(())
    }

    fn select_focused_window(&self) -> Result<()> {
        self.call().unwrap().send(Call::SelectFocusedWindow)?;
        Ok(())
    }

    fn open_fullscreen_window(&self, program_name: String) -> Result<()> {
        self.call()
            .unwrap()
            .send(Call::OpenFullscreenWindow(program_name))?;
        Ok(())
    }

    fn grab_pointer_and_keyboard(&self) -> Result<()> {
        self.call().unwrap().send(Call::GrabPointerAndKeyboard)?;
        Ok(())
    }

    fn unlock_sleep(&self) -> Result<()> {
        self.call().unwrap().send(Call::UnlockSleep)?;
        Ok(())
    }

    fn raise_window(&self) -> Result<()> {
        self.call().unwrap().send(Call::RaiseWindow)?;
        Ok(())
    }

    fn redraw_screen(&self) -> Result<()> {
        self.call().unwrap().send(Call::RedrawScreen)?;
        Ok(())
    }

    fn flush(&self) -> Result<()> {
        self.call().unwrap().send(Call::FlushCommands)?;
        Ok(())
    }
}