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
|
// Copyright 2025-2026 Dillution <[email protected]>.
//
// This file is part of DPIBreak.
//
// DPIBreak 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 3 of the License, or (at your
// option) any later version.
//
// DPIBreak 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 DPIBreak. If not, see <https://www.gnu.org/licenses/>.
use anyhow::{Result, Context};
use std::sync::{
atomic::{Ordering, AtomicBool},
};
mod platform;
mod pkt;
mod tls;
mod log;
mod opt;
use log::LogLevel;
const PROJECT_NAME: &str = "DPIBreak";
const PKG_VERSION: &str = env!("CARGO_PKG_VERSION");
const PKG_DESCRIPTION: &str = env!("CARGO_PKG_DESCRIPTION");
const PKG_HOMEPAGE: &str = env!("CARGO_PKG_HOMEPAGE");
const MESSAGE_AT_RUN: &str = r#"DPIBreak is now running.
Press Ctrl+C or close this window to stop.
"#;
static RUNNING: AtomicBool = AtomicBool::new(true);
fn trap_exit() -> Result<()> {
ctrlc::set_handler(|| {
RUNNING.store(false, Ordering::SeqCst);
}).context("handler: ")?;
Ok(())
}
/// Drop with calling platform::cleanup()
struct EnsureCleanup;
impl Drop for EnsureCleanup {
fn drop(&mut self) {
if let Err(e) = platform::cleanup() {
log_println!(LogLevel::Error, "cleanup failed: {e}");
}
}
}
fn splash_banner() {
splash!("{PROJECT_NAME} v{PKG_VERSION} - {PKG_DESCRIPTION}");
splash!("{PKG_HOMEPAGE}");
splash!("");
}
fn main_0() -> Result<()> {
trap_exit()?;
let opt = opt::Opt::from_args()?;
let initialized = opt.set_opt()?;
splash_banner();
platform::bootstrap()?;
initialized.log();
let _guard = EnsureCleanup;
platform::run()?;
Ok(())
}
fn main() {
let code = match main_0() {
Ok(()) => 0,
Err(e) => {
log_println!(LogLevel::Error, "{e}");
for (i, cause) in e.chain().skip(1).enumerate() {
log_println!(LogLevel::Error, "caused by[{i}]: {cause}");
}
1
}
};
std::process::exit(code);
}
|
