From: Joel Fernandes <joelagn...@nvidia.com>
This will be used in the nova-core driver where we need to upward-align
the image size to get to the next image in the VBIOS ROM.
[acour...@nvidia.com: handled conflicts due to removal of patch creating
num.rs]
Signed-off-by: Joel Fernandes <joelagn...@nvidia.com>
Signed-off-by: Alexandre Courbot <acour...@nvidia.com>
---
rust/kernel/lib.rs | 1 +
rust/kernel/num.rs | 44 ++++++++++++++++++++++++++++++++++++++++++++
2 files changed, 45 insertions(+)
diff --git a/rust/kernel/lib.rs b/rust/kernel/lib.rs
index
ab0286857061d2de1be0279cbd2cd3490e5a48c3..be75b196aa7a29cf3eed7c902ed8fb98689bbb50
100644
--- a/rust/kernel/lib.rs
+++ b/rust/kernel/lib.rs
@@ -67,6 +67,7 @@
pub mod miscdevice;
#[cfg(CONFIG_NET)]
pub mod net;
+pub mod num;
pub mod of;
pub mod page;
#[cfg(CONFIG_PCI)]
diff --git a/rust/kernel/num.rs b/rust/kernel/num.rs
new file mode 100644
index
0000000000000000000000000000000000000000..7d69cbd957546ab75fcf677b851f102c3a5e5c39
--- /dev/null
+++ b/rust/kernel/num.rs
@@ -0,0 +1,44 @@
+// SPDX-License-Identifier: GPL-2.0
+
+//! Numerical and binary utilities for primitive types.
+
+/// A trait providing alignment operations for `usize`.
+use core::ops::{Add, BitAnd, BitOr, Not, Sub};
+
+/// Traits for unsigned integers
+pub trait Unsigned:
+ Copy
+ + BitAnd<Output = Self>
+ + BitOr<Output = Self>
+ + Not<Output = Self>
+ + Add<Output = Self>
+ + Sub<Output = Self>
+ + From<u8>
+{
+}
+
+macro_rules! unsigned_trait_impl {
+ ($($t:ty),+) => {
+ $(
+ impl Unsigned for $t {}
+ )+
+ };
+}
+unsigned_trait_impl!(usize, u8, u16, u32, u64, u128);
+
+/// Trait for unsigned integer alignment
+pub trait NumAlign {
+ /// Implement upward power-of-2 alignment for unsigned ints
+ fn align_up(self, alignment: Self) -> Self;
+}
+
+impl<T> NumAlign for T
+where
+ T: Unsigned,
+{
+ #[inline]
+ fn align_up(self, alignment: Self) -> Self {
+ let one = T::from(1u8);
+ (self + alignment - one) & !(alignment - one)
+ }
+}
--
2.49.0
