alioth/src/utils/utils.rs - a0deb75

utils.rs35.00%

// Copyright 2024 Google LLC
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     https://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
 
pub mod endian;
#[cfg(target_os = "linux")]
pub mod uds;
 
use std::sync::atomic::{AtomicU64, Ordering};
 
pub fn truncate_u64(val: u64, size: u64) -> u64 {39x
    val & (u64::MAX >> (64 - (size << 3)))39x
}39x
 
#[macro_export]
macro_rules! align_up {
    ($num:expr, $bits:expr) => {{
        let mask = (1 << $bits) - 1;
        ($num.wrapping_add(mask)) & !mask
    }};
}
 
#[macro_export]
macro_rules! align_up_ty {
    ($num:expr, $ty:ty) => {{
        let mask = ::core::mem::align_of::<$ty>() - 1;
        ($num.wrapping_add(mask)) & !mask
    }};
}
 
#[macro_export]
macro_rules! align_down {
    ($num:expr, $bits:expr) => {{
        let mask = (1 << $bits) - 1;
        $num & !mask
    }};
}
 
#[macro_export]
macro_rules! assign_bits {
    ($dst:expr, $src:expr, $mask:expr) => {
        $dst = ($dst & !$mask) | ($src & $mask)
    };
}
 
#[macro_export]
macro_rules! mask_bits {
    ($dst:expr, $src:expr, $mask:expr) => {
        ($dst & !$mask) | ($src & $mask)
    };
}
 
#[cfg(target_arch = "x86_64")]
#[inline]
pub fn wrapping_sum<'a, T>(data: T) -> u8
where
    T: IntoIterator<Item = &'a u8>,
{
    data.into_iter().fold(0u8, |accu, e| accu.wrapping_add(*e))
}
 
pub fn get_atomic_low32(num: &AtomicU64) -> u32 {
    num.load(Ordering::Acquire) as u32
}
 
pub fn get_atomic_high32(num: &AtomicU64) -> u32 {
    (num.load(Ordering::Acquire) >> 32) as u32
}
 
pub fn set_low32(num: &mut u64, val: u32) {
    *num &= !0xffff_ffff;
    *num |= val as u64;
}
 
pub fn set_high32(num: &mut u64, val: u32) {
    *num &= 0xffff_ffff;
    *num |= (val as u64) << 32;
}
 
pub fn set_atomic_low32(num: &AtomicU64, val: u32) {
    let mut cur = num.load(Ordering::Acquire);
    set_low32(&mut cur, val);
    num.store(cur, Ordering::Release)
}
 
pub fn set_atomic_high32(num: &AtomicU64, val: u32) {
    let mut cur = num.load(Ordering::Acquire);
    set_high32(&mut cur, val);
    num.store(cur, Ordering::Release)
}
 
#[macro_export]
macro_rules! ffi {
    ($f:expr) => {{
        let ret = $f;
        if ret <= -1 {
            Err(::std::io::Error::last_os_error())
        } else {
            Ok(ret)
        }
    }};
    ($f:expr, $failure:expr) => {{
        let ret = $f;
        if ret == $failure {
            Err(::std::io::Error::last_os_error())
        } else {
            Ok(ret)
        }
    }};
}
 
#[macro_export]
macro_rules! consts {
    (
        $(#[$attr:meta])*
        $vs:vis struct $EnumName:ident($TyName:ty) {
            $( $(#[$vattr:meta])* $VARIANT:ident = $value:expr;)*
        }
    ) => {
        #[repr(transparent)]
        #[derive(PartialEq, Eq, Copy, Clone)]
        $(#[$attr])*
        $vs struct $EnumName($TyName);
 
        impl $EnumName {
            $($(#[$vattr])* pub const $VARIANT: $EnumName = $EnumName($value);)*
 
            #[allow(dead_code)]
            pub const fn raw(self) -> $TyName {104x
                self.0104x
            }104x
 
            #[allow(dead_code)]
            pub const fn name(self) -> &'static str {
                match self {
                    $($EnumName::$VARIANT => stringify!($VARIANT),)*
                    #[allow(unreachable_patterns)]
                    _ => "Unknown"
                }
            }
        }
 
        impl ::core::fmt::Debug for $EnumName {
            fn fmt(&self, f: &mut ::core::fmt::Formatter<'_>) -> ::core::fmt::Result {21x
                f.write_str(stringify!($EnumName))?;21x
                match *self {21x
                    $($EnumName::$VARIANT => {
                        f.write_str("::")?;
                        f.write_str(stringify!($VARIANT))
                    })*
                    #[allow(unreachable_patterns)]
                    _ => {
                        ::core::fmt::Write::write_char(f, '(')?;
                        ::core::fmt::Debug::fmt(&self.0, f)?;
                        ::core::fmt::Write::write_char(f, ')')
                    }
                }
            }21x
        }
 
 
        impl From<$EnumName> for $TyName {
            fn from(value: $EnumName) -> Self {13x
                value.013x
            }13x
        }
 
        impl From<$TyName> for $EnumName {
            fn from(value: $TyName) -> Self {22x
                $EnumName(value)22x
            }22x
        }
    }
}
 
#[macro_export]
macro_rules! bitflags {
    (
        $(#[$attr:meta])*
        $vs:vis struct $FlagTy:ident($TyName:ty) {
            $(
                $(#[$inner:ident $($args:tt)*])*
                $FALG:ident = $value:expr;
            )*
        }
    ) => {
        #[repr(transparent)]
        #[derive(PartialEq, Eq, Copy, Clone, Hash)]
        $(#[$attr])*
        $vs struct $FlagTy($TyName);
 
        ::bitflags::bitflags! {
            impl $FlagTy: $TyName {
                $(
                    $(#[$inner $($args)*])*
                    const $FALG = $value;
                )*
            }
        }
 
        impl ::core::fmt::Debug for $FlagTy {
            fn fmt(&self, f: &mut ::core::fmt::Formatter) -> ::core::fmt::Result {126x
                if self.is_empty() {126x
                    write!(f, "0")51x
                } else {
                    ::bitflags::parser::to_writer(self, f)75x
                }
            }126x
        }
    };
}
 
#[cfg(test)]
#[path = "utils_test.rs"]
mod tests;
 