rust: helpers: Generify the definitions of rust_helper_*_{read,set}*

To support atomic pointers, more {read,set} helpers will be introduced,
hence define macros to generate these helpers to ease the introduction
of the future helpers.

Signed-off-by: Boqun Feng <boqun.feng@gmail.com>
Signed-off-by: Peter Zijlstra (Intel) <peterz@infradead.org>
Link: https://patch.msgid.link/20260117122243.24404-2-boqun.feng@gmail.com
Link: https://patch.msgid.link/20260303201701.12204-4-boqun@kernel.org

diff --git a/rust/helpers/atomic_ext.c b/rust/helpers/atomic_ext.c
index 7d0c2bd340daa10cde2a9cc01ef114f7a7b0b932..f471c1ff123d573a6371e8c605f9fb337067a2e4 100644 (file)
--- a/rust/helpers/atomic_ext.c
+++ b/rust/helpers/atomic_ext.c
@@ -4,45 +4,38 @@
 #include <asm/rwonce.h>
 #include <linux/atomic.h>
 
-__rust_helper s8 rust_helper_atomic_i8_read(s8 *ptr)
-{
-       return READ_ONCE(*ptr);
-}
-
-__rust_helper s8 rust_helper_atomic_i8_read_acquire(s8 *ptr)
-{
-       return smp_load_acquire(ptr);
-}
-
-__rust_helper s16 rust_helper_atomic_i16_read(s16 *ptr)
-{
-       return READ_ONCE(*ptr);
+#define GEN_READ_HELPER(tname, type)                                           \
+__rust_helper type rust_helper_atomic_##tname##_read(type *ptr)                        \
+{                                                                              \
+       return READ_ONCE(*ptr);                                                 \
 }
 
-__rust_helper s16 rust_helper_atomic_i16_read_acquire(s16 *ptr)
-{
-       return smp_load_acquire(ptr);
+#define GEN_SET_HELPER(tname, type)                                            \
+__rust_helper void rust_helper_atomic_##tname##_set(type *ptr, type val)       \
+{                                                                              \
+       WRITE_ONCE(*ptr, val);                                                  \
 }
 
-__rust_helper void rust_helper_atomic_i8_set(s8 *ptr, s8 val)
-{
-       WRITE_ONCE(*ptr, val);
+#define GEN_READ_ACQUIRE_HELPER(tname, type)                                   \
+__rust_helper type rust_helper_atomic_##tname##_read_acquire(type *ptr)                \
+{                                                                              \
+       return smp_load_acquire(ptr);                                           \
 }
 
-__rust_helper void rust_helper_atomic_i8_set_release(s8 *ptr, s8 val)
-{
-       smp_store_release(ptr, val);
+#define GEN_SET_RELEASE_HELPER(tname, type)                                    \
+__rust_helper void rust_helper_atomic_##tname##_set_release(type *ptr, type val)\
+{                                                                              \
+       smp_store_release(ptr, val);                                            \
 }
 
-__rust_helper void rust_helper_atomic_i16_set(s16 *ptr, s16 val)
-{
-       WRITE_ONCE(*ptr, val);
-}
+#define GEN_READ_SET_HELPERS(tname, type)                                      \
+       GEN_READ_HELPER(tname, type)                                            \
+       GEN_SET_HELPER(tname, type)                                             \
+       GEN_READ_ACQUIRE_HELPER(tname, type)                                    \
+       GEN_SET_RELEASE_HELPER(tname, type)                                     \
 
-__rust_helper void rust_helper_atomic_i16_set_release(s16 *ptr, s16 val)
-{
-       smp_store_release(ptr, val);
-}
+GEN_READ_SET_HELPERS(i8, s8)
+GEN_READ_SET_HELPERS(i16, s16)
 
 /*
  * xchg helpers depend on ARCH_SUPPORTS_ATOMIC_RMW and on the