mirror of
https://github.com/EFForg/rayhunter.git
synced 2026-05-06 03:49:08 -07:00
minor nitpicks
This commit is contained in:
24
src/main.rs
24
src/main.rs
@@ -2,29 +2,21 @@ use std::fs::File;
|
||||
use std::mem;
|
||||
use std::os::unix::io::AsRawFd;
|
||||
|
||||
fn main() {
|
||||
const DIAG_IOCTL_SWITCH_LOGGING: u32 = 7;
|
||||
fn main() -> std::io::Result<()> {
|
||||
const DIAG_IOCTL_SWITCH_LOGGING: u32 = 7;
|
||||
const MEMORY_DEVICE_MODE: i32 = 2;
|
||||
const DIAG_IOCTL_REMOTE_DEV: u32 = 32;
|
||||
|
||||
let mut mode_param: [i32; 3] = [MEMORY_DEVICE_MODE, -1, 0]; // diag_logging_mode_param_t
|
||||
let use_mdm: i32 = 0;
|
||||
|
||||
let diag_fd: i32;
|
||||
|
||||
println!("Initializing DIAG");
|
||||
|
||||
let diag_result = File::options().read(true).write(true).open("/dev/diag");
|
||||
match &diag_result {
|
||||
Ok(file) => {
|
||||
diag_fd = file.as_raw_fd();
|
||||
}
|
||||
Err(_) => {
|
||||
println!("error opening diag device.");
|
||||
std::process::exit(1);
|
||||
}
|
||||
}
|
||||
|
||||
let diag_file = File::options()
|
||||
.read(true)
|
||||
.write(true)
|
||||
.open("/dev/diag")?;
|
||||
let diag_fd = diag_file.as_raw_fd();
|
||||
|
||||
unsafe {
|
||||
if libc::ioctl(diag_fd, DIAG_IOCTL_SWITCH_LOGGING, MEMORY_DEVICE_MODE, 0, 0, 0) < 0
|
||||
@@ -40,7 +32,6 @@ fn main() {
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
unsafe {
|
||||
if libc::ioctl(diag_fd, DIAG_IOCTL_REMOTE_DEV, &use_mdm as *const i32) < 0 {
|
||||
println!("ioctl failed 2");
|
||||
@@ -50,4 +41,5 @@ fn main() {
|
||||
|
||||
println!("successfully opened diag device");
|
||||
|
||||
Ok(())
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user