kernel: Rewrite the kernel source code (#554)
* clean unused header * on_module_mounted in ksud.c * refact: use app_profile * unified hook manager * add zygote to hook target * move reboot hook to supercall.c * refactor: kernel_umount setuid_hook * update mark rules, add init mark tracker * remove reboot from check_syscall_fastpath * update setuid_hook, remove uneeded sucompat enable * log freely * kernel: Migrate kprobe hook configuration items * kernel: fix build * cli: add ksud debug mark * Fix rustfmt warning --------- Co-authored-by: Ylarod <me@ylarod.cn> Co-authored-by: Wang Han <416810799@qq.com> Co-authored-by: ShirkNeko <109797057+ShirkNeko@users.noreply.github.com>
This commit is contained in:
@@ -1,9 +1,11 @@
|
||||
use anyhow::{Context, Ok, Result, ensure};
|
||||
use anyhow::{Context, Ok, Result, bail, ensure};
|
||||
use std::{
|
||||
path::{Path, PathBuf},
|
||||
process::Command,
|
||||
};
|
||||
|
||||
use crate::ksucalls;
|
||||
|
||||
const KERNEL_PARAM_PATH: &str = "/sys/module/kernelsu";
|
||||
|
||||
fn read_u32(path: &PathBuf) -> Result<u32> {
|
||||
@@ -50,3 +52,47 @@ pub fn set_manager(pkg: &str) -> Result<()> {
|
||||
let _ = Command::new("am").args(["force-stop", pkg]).status();
|
||||
Ok(())
|
||||
}
|
||||
|
||||
/// Get mark status for a process
|
||||
pub fn mark_get(pid: i32) -> Result<()> {
|
||||
let result = ksucalls::mark_get(pid)?;
|
||||
if pid == 0 {
|
||||
bail!("Please specify a pid to get its mark status");
|
||||
} else {
|
||||
println!(
|
||||
"Process {} mark status: {}",
|
||||
pid,
|
||||
if result != 0 { "marked" } else { "unmarked" }
|
||||
);
|
||||
}
|
||||
Ok(())
|
||||
}
|
||||
|
||||
/// Mark a process
|
||||
pub fn mark_set(pid: i32) -> Result<()> {
|
||||
ksucalls::mark_set(pid)?;
|
||||
if pid == 0 {
|
||||
println!("All processes marked successfully");
|
||||
} else {
|
||||
println!("Process {} marked successfully", pid);
|
||||
}
|
||||
Ok(())
|
||||
}
|
||||
|
||||
/// Unmark a process
|
||||
pub fn mark_unset(pid: i32) -> Result<()> {
|
||||
ksucalls::mark_unset(pid)?;
|
||||
if pid == 0 {
|
||||
println!("All processes unmarked successfully");
|
||||
} else {
|
||||
println!("Process {} unmarked successfully", pid);
|
||||
}
|
||||
Ok(())
|
||||
}
|
||||
|
||||
/// Refresh mark for all running processes
|
||||
pub fn mark_refresh() -> Result<()> {
|
||||
ksucalls::mark_refresh()?;
|
||||
println!("Refreshed mark for all running processes");
|
||||
Ok(())
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user