git: 83412843682b - stable/14 - LinuxKPI: update rcu_dereference_*() and lockdep_is_held()

From: Bjoern A. Zeeb <bz_at_FreeBSD.org>
Date: Thu, 31 Oct 2024 14:34:56 UTC
The branch stable/14 has been updated by bz:

URL: https://cgit.FreeBSD.org/src/commit/?id=83412843682beccc5306cca836160e6e509c563e

commit 83412843682beccc5306cca836160e6e509c563e
Author:     Bjoern A. Zeeb <bz@FreeBSD.org>
AuthorDate: 2024-09-29 00:35:04 +0000
Commit:     Bjoern A. Zeeb <bz@FreeBSD.org>
CommitDate: 2024-10-31 14:33:57 +0000

    LinuxKPI: update rcu_dereference_*() and lockdep_is_held()
    
    Update rcu_dereference_{check,protected}() to call the check and log
    once if it fails and if the RCU debug sysctl is turned on.
    Also add proper checks for conditions passed in to these functions.
    For that implement linux_rcu_read_lock_held() (lots of help from wulf).
    
    (While here also remove extraneous extern for function prototypes).
    
    Update lockdep_is_held() to always be an inline function with argument
    annotation so that we do no longer have unused variables
    in callers which only call lockdep_is_held().
    
    Sponsored by:   The FreeBSD Foundation
    Reviewed by:    wulf
    Differential Revision:  https://reviews.freebsd.org/D46842
    
    (cherry picked from commit 5c92f84bb607c692ce4fa762a7a3c4b86a2fa281)
---
 sys/compat/linuxkpi/common/include/linux/lockdep.h | 28 ++++++-----
 .../linuxkpi/common/include/linux/rcupdate.h       | 57 ++++++++++++++++++----
 sys/compat/linuxkpi/common/src/linux_compat.c      |  4 ++
 sys/compat/linuxkpi/common/src/linux_rcu.c         | 31 ++++++++++++
 4 files changed, 97 insertions(+), 23 deletions(-)

diff --git a/sys/compat/linuxkpi/common/include/linux/lockdep.h b/sys/compat/linuxkpi/common/include/linux/lockdep.h
index 5291be79218e..70a4e45867fe 100644
--- a/sys/compat/linuxkpi/common/include/linux/lockdep.h
+++ b/sys/compat/linuxkpi/common/include/linux/lockdep.h
@@ -70,17 +70,6 @@ struct pin_cookie {
 
 #define	lockdep_assert_none_held_once() do { } while (0)
 
-static __inline bool
-lockdep_is_held(void *__m)
-{
-	struct lock_object *__lock;
-	struct thread *__td;
-
-	__lock = __m;
-	return (LOCK_CLASS(__lock)->lc_owner(__lock, &__td) != 0);
-}
-#define	lockdep_is_held_type(_m, _t) lockdep_is_held(_m)
-
 #else
 #define	lockdep_assert(cond) do { } while (0)
 #define	lockdep_assert_once(cond) do { } while (0)
@@ -91,10 +80,23 @@ lockdep_is_held(void *__m)
 
 #define	lockdep_assert_held_once(m) do { (void)(m); } while (0)
 
-#define	lockdep_is_held(m)	1
-#define	lockdep_is_held_type(_m, _t)	1
 #endif
 
+static __inline bool
+lockdep_is_held(void *__m __diagused)
+{
+#ifdef INVARIANTS
+	struct lock_object *__lock;
+	struct thread *__td;
+
+	__lock = __m;
+	return (LOCK_CLASS(__lock)->lc_owner(__lock, &__td) != 0);
+#else
+	return (true);
+#endif
+}
+#define	lockdep_is_held_type(_m, _t)	lockdep_is_held(_m)
+
 #define	might_lock(m)	do { } while (0)
 #define	might_lock_read(m) do { } while (0)
 #define	might_lock_nested(m, n) do { } while (0)
diff --git a/sys/compat/linuxkpi/common/include/linux/rcupdate.h b/sys/compat/linuxkpi/common/include/linux/rcupdate.h
index 95332217f8f5..85d766c8dbc9 100644
--- a/sys/compat/linuxkpi/common/include/linux/rcupdate.h
+++ b/sys/compat/linuxkpi/common/include/linux/rcupdate.h
@@ -1,6 +1,10 @@
 /*-
  * Copyright (c) 2016-2017 Mellanox Technologies, Ltd.
  * All rights reserved.
+ * Copyright (c) 2024 The FreeBSD Foundation
+ *
+ * Portions of this software were developed by Björn Zeeb
+ * under sponsorship from the FreeBSD Foundation.
  *
  * Redistribution and use in source and binary forms, with or without
  * modification, are permitted provided that the following conditions
@@ -26,11 +30,20 @@
 #ifndef	_LINUXKPI_LINUX_RCUPDATE_H_
 #define	_LINUXKPI_LINUX_RCUPDATE_H_
 
+#include <sys/cdefs.h>
+
 #include <linux/compiler.h>
 #include <linux/types.h>
+#include <linux/kernel.h>
 
 #include <machine/atomic.h>
 
+extern int linuxkpi_rcu_debug;
+#define	RCU_WARN_ONCE(c, ...)	do {					\
+	if (unlikely(linuxkpi_rcu_debug > 0))				\
+		WARN_ONCE((c), ##__VA_ARGS__);				\
+} while(0)
+
 #define	LINUX_KFREE_RCU_OFFSET_MAX	4096	/* exclusive */
 
 /* BSD specific defines */
@@ -61,6 +74,9 @@
 	linux_rcu_read_unlock(RCU_TYPE_REGULAR);\
 } while (0)
 
+#define	rcu_read_lock_held(void)					\
+	linux_rcu_read_lock_held(RCU_TYPE_REGULAR)
+
 #define	synchronize_rcu(void) do {	\
 	linux_synchronize_rcu(RCU_TYPE_REGULAR);	\
 } while (0)
@@ -79,14 +95,34 @@
 #define	rcu_access_pointer(p)			\
 	((__typeof(*p) *)READ_ONCE(p))
 
-#define	rcu_dereference_protected(p, c)		\
+#define	rcu_dereference(p)			\
 	((__typeof(*p) *)READ_ONCE(p))
 
-#define	rcu_dereference(p)			\
-	rcu_dereference_protected(p, 0)
+#define	__rcu_var_name(n, f, l)						\
+	__CONCAT(__CONCAT(__CONCAT(rcu_, n), _), __COUNTER__)
+
+#define	__rcu_dereference_protected(p, c, n)				\
+({									\
+    RCU_WARN_ONCE(!(c), "%s:%d: condition for %s failed\n",		\
+	__func__, __LINE__, __XSTRING(n));				\
+    rcu_dereference(p);							\
+})
+
+#define	rcu_dereference_protected(p, c)					\
+    __rcu_dereference_protected((p), (c),				\
+	__rcu_var_name(protected, __func__, __LINE__))
+
+#define	__rcu_dereference_check(p, c, n)				\
+({									\
+    __typeof(*p) *n = rcu_dereference(p);				\
+    RCU_WARN_ONCE(!(c), "%s:%d: condition for %s failed\n",		\
+	__func__, __LINE__, __XSTRING(n));				\
+    n;									\
+})
 
-#define	rcu_dereference_check(p, c)		\
-	rcu_dereference_protected(p, c)
+#define	rcu_dereference_check(p, c)					\
+    __rcu_dereference_check((p), (c) || rcu_read_lock_held(),		\
+	__rcu_var_name(check, __func__, __LINE__))
 
 #define	rcu_dereference_raw(p)			\
 	((__typeof(*p) *)READ_ONCE(p))
@@ -113,11 +149,12 @@
 
 /* prototypes */
 
-extern void linux_call_rcu(unsigned type, struct rcu_head *ptr, rcu_callback_t func);
-extern void linux_rcu_barrier(unsigned type);
-extern void linux_rcu_read_lock(unsigned type);
-extern void linux_rcu_read_unlock(unsigned type);
-extern void linux_synchronize_rcu(unsigned type);
+void linux_call_rcu(unsigned type, struct rcu_head *ptr, rcu_callback_t func);
+void linux_rcu_barrier(unsigned type);
+void linux_rcu_read_lock(unsigned type);
+void linux_rcu_read_unlock(unsigned type);
+bool linux_rcu_read_lock_held(unsigned);
+void linux_synchronize_rcu(unsigned type);
 
 /* Empty implementation for !DEBUG */
 #define	init_rcu_head(...)
diff --git a/sys/compat/linuxkpi/common/src/linux_compat.c b/sys/compat/linuxkpi/common/src/linux_compat.c
index ec4e59f66e7f..fe1a545c6a3a 100644
--- a/sys/compat/linuxkpi/common/src/linux_compat.c
+++ b/sys/compat/linuxkpi/common/src/linux_compat.c
@@ -117,6 +117,10 @@ int linuxkpi_debug;
 SYSCTL_INT(_compat_linuxkpi, OID_AUTO, debug, CTLFLAG_RWTUN,
     &linuxkpi_debug, 0, "Set to enable pr_debug() prints. Clear to disable.");
 
+int linuxkpi_rcu_debug;
+SYSCTL_INT(_compat_linuxkpi, OID_AUTO, rcu_debug, CTLFLAG_RWTUN,
+    &linuxkpi_rcu_debug, 0, "Set to enable RCU warning. Clear to disable.");
+
 int linuxkpi_warn_dump_stack = 0;
 SYSCTL_INT(_compat_linuxkpi, OID_AUTO, warn_dump_stack, CTLFLAG_RWTUN,
     &linuxkpi_warn_dump_stack, 0,
diff --git a/sys/compat/linuxkpi/common/src/linux_rcu.c b/sys/compat/linuxkpi/common/src/linux_rcu.c
index 4515684aabc7..8151d7fd15e1 100644
--- a/sys/compat/linuxkpi/common/src/linux_rcu.c
+++ b/sys/compat/linuxkpi/common/src/linux_rcu.c
@@ -2,6 +2,10 @@
  * Copyright (c) 2016 Matthew Macy (mmacy@mattmacy.io)
  * Copyright (c) 2017-2021 Hans Petter Selasky (hselasky@freebsd.org)
  * All rights reserved.
+ * Copyright (c) 2024 The FreeBSD Foundation
+ *
+ * Portions of this software were developed by Björn Zeeb
+ * under sponsorship from the FreeBSD Foundation.
  *
  * Redistribution and use in source and binary forms, with or without
  * modification, are permitted provided that the following conditions
@@ -246,6 +250,33 @@ linux_rcu_read_unlock(unsigned type)
 	sched_unpin();
 }
 
+bool
+linux_rcu_read_lock_held(unsigned type)
+{
+#ifdef INVARINATS
+	struct linux_epoch_record *record __diagused;
+	struct task_struct *ts;
+
+	MPASS(type < RCU_TYPE_MAX);
+
+	if (RCU_SKIP())
+		return (false);
+
+	if (__current_unallocated(curthread))
+		return (false);
+
+	ts = current;
+	if (ts->rcu_recurse[type] == 0)
+		return (false);
+
+	MPASS(curthread->td_pinned != 0);
+	MPASS((record = &DPCPU_GET(linux_epoch_record[type])) &&
+	    record->epoch_record.active != 0);
+#endif
+
+	return (true);
+}
+
 static void
 linux_synchronize_rcu_cb(ck_epoch_t *epoch __unused, ck_epoch_record_t *epoch_record, void *arg __unused)
 {