0baccb7621
Co-authored-by: backslashxx <118538522+backslashxx@users.noreply.github.com> Co-authored-by: SChernykh <15806605+SChernykh@users.noreply.github.com> Co-authored-by: ShirkNeko <109797057+ShirkNeko@users.noreply.github.com> Signed-off-by: ShirkNeko <109797057+ShirkNeko@users.noreply.github.com>
56 lines
1.7 KiB
Rust
56 lines
1.7 KiB
Rust
use anyhow::Result;
|
|
use const_format::concatcp;
|
|
use rust_embed::RustEmbed;
|
|
use std::path::Path;
|
|
|
|
use crate::{defs::BINARY_DIR, utils};
|
|
|
|
pub const RESETPROP_PATH: &str = concatcp!(BINARY_DIR, "resetprop");
|
|
pub const BUSYBOX_PATH: &str = concatcp!(BINARY_DIR, "busybox");
|
|
pub const BOOTCTL_PATH: &str = concatcp!(BINARY_DIR, "bootctl");
|
|
|
|
#[cfg(all(target_arch = "x86_64", target_os = "android"))]
|
|
#[derive(RustEmbed)]
|
|
#[folder = "bin/x86_64"]
|
|
struct Asset;
|
|
|
|
// IF NOT x86_64 ANDROID, ie. macos, linux, windows, always use aarch64
|
|
#[cfg(all(target_arch = "aarch64", target_os = "android"))]
|
|
#[derive(RustEmbed)]
|
|
#[folder = "bin/aarch64"]
|
|
struct Asset;
|
|
|
|
#[cfg(all(target_arch = "arm", target_os = "android"))]
|
|
#[derive(RustEmbed)]
|
|
#[folder = "bin/arm"]
|
|
struct Asset;
|
|
|
|
pub fn ensure_binaries(ignore_if_exist: bool) -> Result<()> {
|
|
for file in Asset::iter() {
|
|
if file == "ksuinit" || file.ends_with(".ko") {
|
|
// don't extract ksuinit and kernel modules
|
|
continue;
|
|
}
|
|
let asset = Asset::get(&file).ok_or(anyhow::anyhow!("asset not found: {}", file))?;
|
|
utils::ensure_binary(format!("{BINARY_DIR}{file}"), &asset.data, ignore_if_exist)?
|
|
}
|
|
Ok(())
|
|
}
|
|
|
|
pub fn copy_assets_to_file(name: &str, dst: impl AsRef<Path>) -> Result<()> {
|
|
let asset = Asset::get(name).ok_or(anyhow::anyhow!("asset not found: {}", name))?;
|
|
std::fs::write(dst, asset.data)?;
|
|
Ok(())
|
|
}
|
|
|
|
pub fn list_supported_kmi() -> Result<Vec<String>> {
|
|
let mut list = Vec::new();
|
|
for file in Asset::iter() {
|
|
// kmi_name = "xxx_kernelsu.ko"
|
|
if let Some(kmi) = file.strip_suffix("_kernelsu.ko") {
|
|
list.push(kmi.to_string());
|
|
}
|
|
}
|
|
Ok(list)
|
|
}
|