android/SukiSU-Ultra
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

ksud: don't reclaim when enable/disable modules

Browse Source
This commit is contained in:
weishu
2024-02-19 16:36:41 +08:00
parent e0e7058d14
commit cfc982f2f3
3 changed files with 24 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
userspace/ksud/src/module.rs
View File

@@ -475,7 +475,7 @@ pub fn install_module(zip: &str) -> Result<()> {
result
}
fn update_module<F>(update_dir: &str, id: &str, func: F) -> Result<()>
fn update_module<F>(update_dir: &str, id: &str, punch_hole: bool, func: F) -> Result<()>
where
F: Fn(&str, &str) -> Result<()>,
{
@@ -511,8 +511,10 @@ where
// call the operation func
let result = func(id, update_dir);
if let Err(e) = utils::punch_hole(modules_update_tmp_img) {
warn!("Failed to punch hole: {}", e);
if punch_hole {
if let Err(e) = utils::punch_hole(modules_update_tmp_img) {
warn!("Failed to punch hole: {}", e);
}
}
if let Err(e) = std::fs::rename(modules_update_tmp_img, defs::MODULE_UPDATE_IMG) {
@@ -528,7 +530,7 @@ where
}
pub fn uninstall_module(id: &str) -> Result<()> {
update_module(defs::MODULE_UPDATE_TMP_DIR, id, |mid, update_dir| {
update_module(defs::MODULE_UPDATE_TMP_DIR, id, true, |mid, update_dir| {
let dir = Path::new(update_dir);
ensure!(dir.exists(), "No module installed");
@@ -594,13 +596,13 @@ fn _enable_module(module_dir: &str, mid: &str, enable: bool) -> Result<()> {
}
pub fn enable_module(id: &str) -> Result<()> {
update_module(defs::MODULE_UPDATE_TMP_DIR, id, |mid, update_dir| {
update_module(defs::MODULE_UPDATE_TMP_DIR, id, false, |mid, update_dir| {
_enable_module(update_dir, mid, true)
})
}
pub fn disable_module(id: &str) -> Result<()> {
update_module(defs::MODULE_UPDATE_TMP_DIR, id, |mid, update_dir| {
update_module(defs::MODULE_UPDATE_TMP_DIR, id, false, |mid, update_dir| {
_enable_module(update_dir, mid, false)
})
}

6
userspace/ksud/src/utils.rs
View File

@@ -232,6 +232,7 @@ pub fn copy_sparse_file<P: AsRef<Path>, Q: AsRef<Path>>(src: P, dst: Q) -> Resul
Ok(())
}
#[cfg(any(target_os = "linux", target_os = "android"))]
pub fn punch_hole(src: impl AsRef<Path>) -> Result<()> {
let mut src_file = OpenOptions::new().write(true).read(true).open(src)?;
@@ -316,3 +317,8 @@ pub fn punch_hole(src: impl AsRef<Path>) -> Result<()> {
Ok(())
}
#[cfg(not(any(target_os = "linux", target_os = "android")))]
pub fn punch_hole(src: impl AsRef<Path>) -> Result<()> {
unimplemented!()
}