62 lines
1.7 KiB
Rust
62 lines
1.7 KiB
Rust
use anyhow::{ensure, Context, Ok, Result};
|
|
use std::{
|
|
path::{Path, PathBuf},
|
|
process::Command,
|
|
};
|
|
|
|
use crate::apk_sign::get_apk_signature;
|
|
|
|
const KERNEL_PARAM_PATH: &str = "/sys/module/kernelsu";
|
|
|
|
fn read_u32(path: &PathBuf) -> Result<u32> {
|
|
let content = std::fs::read_to_string(path)?;
|
|
let content = content.trim();
|
|
let content = content.parse::<u32>()?;
|
|
Ok(content)
|
|
}
|
|
|
|
fn set_kernel_param(size: u32, hash: u32) -> Result<()> {
|
|
let kernel_param_path = Path::new(KERNEL_PARAM_PATH).join("parameters");
|
|
let expeced_size_path = kernel_param_path.join("ksu_expected_size");
|
|
let expeced_hash_path = kernel_param_path.join("ksu_expected_hash");
|
|
|
|
println!(
|
|
"before size: {:#x} hash: {:#x}",
|
|
read_u32(&expeced_size_path)?,
|
|
read_u32(&expeced_hash_path)?
|
|
);
|
|
|
|
std::fs::write(&expeced_size_path, size.to_string())?;
|
|
std::fs::write(&expeced_hash_path, hash.to_string())?;
|
|
|
|
println!(
|
|
"after size: {:#x} hash: {:#x}",
|
|
read_u32(&expeced_size_path)?,
|
|
read_u32(&expeced_hash_path)?
|
|
);
|
|
|
|
Ok(())
|
|
}
|
|
|
|
fn get_apk_path(package_name: &str) -> Result<String> {
|
|
// `cmd package path` is not available below Android 9
|
|
let output = Command::new("pm").args(["path", package_name]).output()?;
|
|
|
|
// package:/data/app/<xxxx>/base.apk
|
|
let output = String::from_utf8_lossy(&output.stdout);
|
|
let path = output.trim().replace("package:", "");
|
|
Ok(path)
|
|
}
|
|
|
|
pub fn set_manager(pkg: &str) -> Result<()> {
|
|
ensure!(
|
|
Path::new(KERNEL_PARAM_PATH).exists(),
|
|
"CONFIG_KSU_DEBUG is not enabled"
|
|
);
|
|
|
|
let path = get_apk_path(pkg).with_context(|| format!("{pkg} does not exist!"))?;
|
|
let sign = get_apk_signature(&path)?;
|
|
set_kernel_param(sign.0, sign.1)?;
|
|
Ok(())
|
|
}
|