summaryrefslogtreecommitdiffstats
path: root/src/main.rs
blob: 5df35241496c7ddaf7c22f7e5bbb59630cb70e07 (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
// -*- coding: utf-8 -*-
//
// pwman-f2e
//
// Copyright 2023 Michael Büsch <m@bues.ch>
//
// This program is free software; you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation; either version 2 of the License, or
// (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License along
// with this program; if not, write to the Free Software Foundation, Inc.,
// 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
//

mod ctap;
mod hid;

use anyhow::{self as ah, Context as _};
use privdrop::PrivDrop;
use std::thread::sleep;
use std::time::Duration;

use crate::ctap::Ctap;
use crate::hid::CtapHid;

fn main() -> ah::Result<()> {
    let mut hid = CtapHid::new().context("Failed to create emulated HID device.")?;
    let mut ctap = Ctap::new();

    // Drop root privileges.
    //TODO config user/group
    PrivDrop::default()
        .user("mb")
        .group("mb")
        .apply()
        .context("Failed to drop privileges.")?;

    loop {
        let mut error = false;

        if let Err(e) = hid.read() {
            eprintln!("HID read error: {e}");
            error = true;
        }
        if !error {
            for (cid, command) in hid.get_commands() {
                match ctap.handle_command(command) {
                    Ok(response) => {
                        if let Err(e) = hid.send_response(cid, response) {
                            eprintln!("HID write error: {e}");
                            error = true;
                        }
                    }
                    Err(e) => {
                        eprintln!("Failed to handle CTAP command: {e}");
                        error = true;
                    }
                }
            }
        }

        if error {
            sleep(Duration::from_millis(500));
        } else {
            sleep(Duration::from_millis(10));
        }
    }
}

// vim: ts=4 sw=4 expandtab
bues.ch cgit interface