#ifndef __XFS_MESSAGE_H
 #define __XFS_MESSAGE_H 1
 
+#include <linux/once_lite.h>
+
 struct xfs_mount;
 
 extern __printf(2, 3)
 } while (0)
 
 #define xfs_printk_once(func, dev, fmt, ...)                   \
-({                                                             \
-       static bool __section(".data.once") __print_once;       \
-       bool __ret_print_once = !__print_once;                  \
-                                                               \
-       if (!__print_once) {                                    \
-               __print_once = true;                            \
-               func(dev, fmt, ##__VA_ARGS__);                  \
-       }                                                       \
-       unlikely(__ret_print_once);                             \
-})
+       DO_ONCE_LITE(func, dev, fmt, ##__VA_ARGS__)
 
 #define xfs_emerg_ratelimited(dev, fmt, ...)                           \
        xfs_printk_ratelimited(xfs_emerg, dev, fmt, ##__VA_ARGS__)
 
 
 #include <linux/compiler.h>
 #include <linux/instrumentation.h>
+#include <linux/once_lite.h>
 
 #define CUT_HERE               "------------[ cut here ]------------\n"
 
 })
 
 #ifndef WARN_ON_ONCE
-#define WARN_ON_ONCE(condition)        ({                              \
-       static bool __section(".data.once") __warned;           \
-       int __ret_warn_once = !!(condition);                    \
-                                                               \
-       if (unlikely(__ret_warn_once && !__warned)) {           \
-               __warned = true;                                \
-               WARN_ON(1);                                     \
-       }                                                       \
-       unlikely(__ret_warn_once);                              \
-})
+#define WARN_ON_ONCE(condition)                                        \
+       DO_ONCE_LITE_IF(condition, WARN_ON, 1)
 #endif
 
-#define WARN_ONCE(condition, format...)        ({                      \
-       static bool __section(".data.once") __warned;           \
-       int __ret_warn_once = !!(condition);                    \
-                                                               \
-       if (unlikely(__ret_warn_once && !__warned)) {           \
-               __warned = true;                                \
-               WARN(1, format);                                \
-       }                                                       \
-       unlikely(__ret_warn_once);                              \
-})
+#define WARN_ONCE(condition, format...)                                \
+       DO_ONCE_LITE_IF(condition, WARN, 1, format)
 
-#define WARN_TAINT_ONCE(condition, taint, format...)   ({      \
-       static bool __section(".data.once") __warned;           \
-       int __ret_warn_once = !!(condition);                    \
-                                                               \
-       if (unlikely(__ret_warn_once && !__warned)) {           \
-               __warned = true;                                \
-               WARN_TAINT(1, taint, format);                   \
-       }                                                       \
-       unlikely(__ret_warn_once);                              \
-})
+#define WARN_TAINT_ONCE(condition, taint, format...)           \
+       DO_ONCE_LITE_IF(condition, WARN_TAINT, 1, taint, format)
 
 #else /* !CONFIG_BUG */
 #ifndef HAVE_ARCH_BUG
 
--- /dev/null
+/* SPDX-License-Identifier: GPL-2.0 */
+#ifndef _LINUX_ONCE_LITE_H
+#define _LINUX_ONCE_LITE_H
+
+#include <linux/types.h>
+
+/* Call a function once. Similar to DO_ONCE(), but does not use jump label
+ * patching via static keys.
+ */
+#define DO_ONCE_LITE(func, ...)                                                \
+       DO_ONCE_LITE_IF(true, func, ##__VA_ARGS__)
+#define DO_ONCE_LITE_IF(condition, func, ...)                          \
+       ({                                                              \
+               static bool __section(".data.once") __already_done;     \
+               bool __ret_do_once = !!(condition);                     \
+                                                                       \
+               if (unlikely(__ret_do_once && !__already_done)) {       \
+                       __already_done = true;                          \
+                       func(__VA_ARGS__);                              \
+               }                                                       \
+               unlikely(__ret_do_once);                                \
+       })
+
+#endif /* _LINUX_ONCE_LITE_H */
 
 #include <linux/linkage.h>
 #include <linux/cache.h>
 #include <linux/ratelimit_types.h>
+#include <linux/once_lite.h>
 
 extern const char linux_banner[];
 extern const char linux_proc_banner[];
 
 #ifdef CONFIG_PRINTK
 #define printk_once(fmt, ...)                                  \
-({                                                             \
-       static bool __section(".data.once") __print_once;       \
-       bool __ret_print_once = !__print_once;                  \
-                                                               \
-       if (!__print_once) {                                    \
-               __print_once = true;                            \
-               printk(fmt, ##__VA_ARGS__);                     \
-       }                                                       \
-       unlikely(__ret_print_once);                             \
-})
+       DO_ONCE_LITE(printk, fmt, ##__VA_ARGS__)
 #define printk_deferred_once(fmt, ...)                         \
-({                                                             \
-       static bool __section(".data.once") __print_once;       \
-       bool __ret_print_once = !__print_once;                  \
-                                                               \
-       if (!__print_once) {                                    \
-               __print_once = true;                            \
-               printk_deferred(fmt, ##__VA_ARGS__);            \
-       }                                                       \
-       unlikely(__ret_print_once);                             \
-})
+       DO_ONCE_LITE(printk_deferred, fmt, ##__VA_ARGS__)
 #else
 #define printk_once(fmt, ...)                                  \
        no_printk(fmt, ##__VA_ARGS__)
 
 #include <linux/irq_work.h>
 #include <linux/workqueue.h>
 #include <linux/ctype.h>
+#include <linux/once_lite.h>
 
 #ifdef CONFIG_FTRACE_SYSCALLS
 #include <asm/unistd.h>                /* For NR_SYSCALLS           */
 #include "trace_entries.h"
 
 /* Use this for memory failure errors */
-#define MEM_FAIL(condition, fmt, ...) ({                       \
-       static bool __section(".data.once") __warned;           \
-       int __ret_warn_once = !!(condition);                    \
-                                                               \
-       if (unlikely(__ret_warn_once && !__warned)) {           \
-               __warned = true;                                \
-               pr_err("ERROR: " fmt, ##__VA_ARGS__);           \
-       }                                                       \
-       unlikely(__ret_warn_once);                              \
-})
+#define MEM_FAIL(condition, fmt, ...)                                  \
+       DO_ONCE_LITE_IF(condition, pr_err, "ERROR: " fmt, ##__VA_ARGS__)
 
 /*
  * syscalls are special, and need special handling, this is why