util: Fix alarm() not found on riscv64

XuShaohua · Oct 12, 2023 · 28bd842 · 28bd842
1 parent b14bf9c
commit 28bd842
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 6 deletions.
diff --git a/src/platform/linux-types/uapi/asm_generic/mod.rs b/src/platform/linux-types/uapi/asm_generic/mod.rs
@@ -2,14 +2,22 @@
 // Use of this source is governed by Apache-2.0 License that can be found
 // in the LICENSE file.
 
-#[cfg(any(target_arch = "aarch64", target_arch = "loongarch64"))]
+#[cfg(any(
+    target_arch = "aarch64",
+    target_arch = "loongarch64",
+    target_arch = "riscv64"
+))]
 mod signal;
-#[cfg(any(target_arch = "aarch64", target_arch = "loongarch64"))]
+#[cfg(any(
+    target_arch = "aarch64",
+    target_arch = "loongarch64",
+    target_arch = "riscv64"
+))]
 pub use signal::*;
 
-#[cfg(test_arch = "loongarch64")]
+#[cfg(any(test_arch = "loongarch64", target_arch = "riscv64"))]
 mod stat;
-#[cfg(test_arch = "loongarch64")]
+#[cfg(any(test_arch = "loongarch64", target_arch = "riscv64"))]
 pub use stat::*;
 
 mod fcntl;

diff --git a/src/util.rs b/src/util.rs
@@ -153,7 +153,8 @@ pub fn alarm(seconds: u32) -> Result<u32, crate::Errno> {
     #[cfg(any(
         target_arch = "aarch64",
         target_arch = "arm",
-        target_arch = "loongarch64"
+        target_arch = "loongarch64",
+        target_arch = "riscv64",
     ))]
     let remaining = {
         let mut it = crate::itimerval_t::default();
@@ -166,7 +167,8 @@ pub fn alarm(seconds: u32) -> Result<u32, crate::Errno> {
     #[cfg(not(any(
         target_arch = "aarch64",
         target_arch = "arm",
-        target_arch = "loongarch64"
+        target_arch = "loongarch64",
+        target_arch = "riscv64",
     )))]
     let remaining = unsafe { crate::alarm(seconds) };