ReZygisk/zygiskd/src/root_impl/mod.rs

mod kernelsu;
mod magisk;

pub enum RootImpl {
    None,
    TooOld,
    Abnormal,
    Multiple,
    KernelSU,
    Magisk,
}

// FIXME: OnceCell bugs on 32 bit
static mut ROOT_IMPL: RootImpl = RootImpl::None;

pub fn setup() {
    let ksu_version = kernelsu::get_kernel_su();
    let magisk_version = magisk::get_magisk();

    let impl_ = match (ksu_version, magisk_version) {
        (None, None) => RootImpl::None,
        (Some(_), Some(_)) => RootImpl::Multiple,
        (Some(ksu_version), None) => {
            match ksu_version {
                kernelsu::Version::Supported => RootImpl::KernelSU,
                kernelsu::Version::TooOld => RootImpl::TooOld,
                kernelsu::Version::Abnormal => RootImpl::Abnormal,
            }
        }
        (None, Some(magisk_version)) => {
            match magisk_version {
                magisk::Version::Supported => RootImpl::Magisk,
                magisk::Version::TooOld => RootImpl::TooOld,
            }
        }
    };
    unsafe { ROOT_IMPL = impl_; }
}

pub fn get_impl() -> &'static RootImpl {
    unsafe { &ROOT_IMPL }
}

// FIXME: Without #[inline(never)], this function will lag forever
#[inline(never)]
pub fn uid_on_allowlist(uid: i32) -> bool {
    match get_impl() {
        RootImpl::KernelSU => kernelsu::uid_on_allowlist(uid),
        RootImpl::Magisk => magisk::uid_on_allowlist(uid),
        _ => unreachable!(),
    }
}

#[inline(never)]
pub fn uid_on_denylist(uid: i32) -> bool {
    match get_impl() {
        RootImpl::KernelSU => kernelsu::uid_on_denylist(uid),
        RootImpl::Magisk => magisk::uid_on_denylist(uid),
        _ => unreachable!(),
    }
}