rust: kernel: Add simple siphash abstraction

This allows Rust code to use the Hasher interface with the kernel
siphash implementation.

Signed-off-by: Asahi Lina <[email protected]>

Author: hoshinolina

rust/bindings/bindings_helper.h

@@ -29,6 +29,7 @@
 #include <linux/platform_device.h>
 #include <linux/refcount.h>
 #include <linux/wait.h>
+#include <linux/siphash.h>
 #include <linux/sched.h>
 #include <linux/soc/apple/rtkit.h>
 #include <linux/timekeeping.h>

rust/helpers.c

@@ -34,6 +34,7 @@
 #include <linux/platform_device.h>
 #include <linux/refcount.h>
 #include <linux/mutex.h>
+#include <linux/siphash.h>
 #include <linux/spinlock.h>
 #include <linux/sched/signal.h>
 #include <linux/wait.h>
@@ -151,6 +152,13 @@ const char *rust_helper_errname(int err)
 }
 EXPORT_SYMBOL_GPL(rust_helper_errname);
 
+u64 rust_helper_siphash(const void *data, size_t len,
+			const siphash_key_t *key)
+{
+	return siphash(data, len, key);
+}
+EXPORT_SYMBOL_GPL(rust_helper_siphash);
+
 void rust_helper_xa_init_flags(struct xarray *xa, gfp_t flags)
 {
 	xa_init_flags(xa, flags);

rust/kernel/lib.rs

@@ -57,6 +57,7 @@ pub mod of;
 pub mod platform;
 pub mod prelude;
 pub mod print;
+pub mod siphash;
 pub mod soc;
 mod static_assert;
 #[doc(hidden)]

rust/kernel/siphash.rs

@@ -0,0 +1,39 @@
+// SPDX-License-Identifier: GPL-2.0
+
+//! A core::hash::Hasher wrapper for the kernel siphash implementation.
+//!
+//! This module allows Rust code to use the kernel's siphash implementation
+//! to hash Rust objects.
+
+use core::hash::Hasher;
+
+/// A Hasher implementation that uses the kernel siphash implementation.
+#[derive(Default)]
+pub struct SipHasher {
+    // SipHash state is 4xu64, but the Linux implementation
+    // doesn't expose incremental hashing so let's just chain
+    // individual SipHash calls for now, which return a u64
+    // hash.
+    state: u64,
+}
+
+impl SipHasher {
+    /// Create a new SipHasher with zeroed state.
+    pub fn new() -> Self {
+        SipHasher { state: 0 }
+    }
+}
+
+impl Hasher for SipHasher {
+    fn finish(&self) -> u64 {
+        self.state
+    }
+
+    fn write(&mut self, bytes: &[u8]) {
+        let key = bindings::siphash_key_t {
+            key: [self.state, 0],
+        };
+
+        self.state = unsafe { bindings::siphash(bytes.as_ptr() as *const _, bytes.len(), &key) };
+    }
+}