smartdns/plugin/smartdns-ui/src/utils.rs

/*************************************************************************
 *
 * Copyright (C) 2018-2025 Ruilin Peng (Nick) <[email protected]>.
 *
 * smartdns is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * smartdns is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

use nix::libc;
use pbkdf2::{
    password_hash::{rand_core::OsRng, PasswordHash, PasswordHasher, PasswordVerifier, SaltString},
    Pbkdf2,
};

pub fn parse_value<T>(value: Option<String>, min: T, max: T, default: T) -> T
where
    T: PartialOrd + std::str::FromStr,
{
    if value.is_none() {
        return default;
    }

    let value = value.unwrap().parse::<T>();
    if let Err(_) = value {
        return default;
    }

    let mut value = value.unwrap_or_else(|_| default);

    if value < min {
        value = min;
    }

    if value > max {
        value = max;
    }

    value
}

pub fn seconds_until_next_hour() -> u64 {
    let now = chrono::Local::now();
    let minutes = chrono::Timelike::minute(&now);
    let seconds = chrono::Timelike::second(&now);
    let remaining_seconds = 3600 - (minutes * 60 + seconds) as u64;
    remaining_seconds
}

pub fn get_free_disk_space(path: &str) -> u64 {
    let path = std::ffi::CString::new(path).unwrap();
    let mut statvfs: libc::statvfs = unsafe { std::mem::zeroed() };
    let ret = unsafe { libc::statvfs(path.as_ptr(), &mut statvfs) };
    if ret != 0 {
        return 0;
    }
    statvfs.f_bsize as u64 * statvfs.f_bavail as u64
}

pub fn hash_password(password: &str, round: Option<u32>) -> Result<String, Box<dyn std::error::Error>> {
    let salt = SaltString::generate(&mut OsRng);
    let mut parm = pbkdf2::Params::default();
    parm.rounds = round.unwrap_or(10000);
    let password_hash = Pbkdf2
        .hash_password_customized(password.as_bytes(), None, None, parm, &salt)
        .map_err(|e| e.to_string())?
        .to_string();
    Ok(password_hash)
}

pub fn verify_password(password: &str, password_hash: &str) -> bool {
    let parsed_hash = match PasswordHash::new(&password_hash) {
        Ok(h) => h,
        Err(_) => return false,
    };

    Pbkdf2
        .verify_password(password.as_bytes(), &parsed_hash)
        .is_ok()
}

pub fn get_page_size() -> usize {
    // Use libc to get the system page size
    unsafe { libc::sysconf(libc::_SC_PAGESIZE) as usize }
}