History

rbtree, rtmutex: Use rb_add_cached()

Reduce rbtree boiler plate by using the new helpers.

Peter Zijlstra committed 6 years ago

Commit 5a798721 parent a905e841 file · +18 −36Browse files

Unified · Split

kernel/locking/rtmutex.c+18−36
@@ -267,27 +267,18 @@ rt_mutex_waiter_equal(struct rt_mutex_waiter *left, · 266 unmodified lines
267267 return 1;
268268 }
269269
270+#define __node_2_waiter(node) \
271+ rb_entry((node), struct rt_mutex_waiter, tree_entry)
272+
273+static inline bool __waiter_less(struct rb_node *a, const struct rb_node *b)
274+{
275+ return rt_mutex_waiter_less(__node_2_waiter(a), __node_2_waiter(b));
276+}
277+
270278 static void
271279 rt_mutex_enqueue(struct rt_mutex *lock, struct rt_mutex_waiter *waiter)
272280 {
273− struct rb_node **link = &lock->waiters.rb_root.rb_node;
274− struct rb_node *parent = NULL;
275− struct rt_mutex_waiter *entry;
276− bool leftmost = true;
277−
278− while (*link) {
279− parent = *link;
280− entry = rb_entry(parent, struct rt_mutex_waiter, tree_entry);
281− if (rt_mutex_waiter_less(waiter, entry)) {
282− link = &parent->rb_left;
283− } else {
284− link = &parent->rb_right;
285− leftmost = false;
286− }
287− }
288−
289− rb_link_node(&waiter->tree_entry, parent, link);
290− rb_insert_color_cached(&waiter->tree_entry, &lock->waiters, leftmost);
281+ rb_add_cached(&waiter->tree_entry, &lock->waiters, __waiter_less);
291282 }
292283
293284 static void
@@ -300,27 +291,18 @@ rt_mutex_dequeue(struct rt_mutex *lock, struct rt_mutex_waiter *waiter) · 6 unmodified lines
300291 RB_CLEAR_NODE(&waiter->tree_entry);
301292 }
302293
294+#define __node_2_pi_waiter(node) \
295+ rb_entry((node), struct rt_mutex_waiter, pi_tree_entry)
296+
297+static inline bool __pi_waiter_less(struct rb_node *a, const struct rb_node *b)
298+{
299+ return rt_mutex_waiter_less(__node_2_pi_waiter(a), __node_2_pi_waiter(b));
300+}
301+
303302 static void
304303 rt_mutex_enqueue_pi(struct task_struct *task, struct rt_mutex_waiter *waiter)
305304 {
306− struct rb_node **link = &task->pi_waiters.rb_root.rb_node;
307− struct rb_node *parent = NULL;
308− struct rt_mutex_waiter *entry;
309− bool leftmost = true;
310−
311− while (*link) {
312− parent = *link;
313− entry = rb_entry(parent, struct rt_mutex_waiter, pi_tree_entry);
314− if (rt_mutex_waiter_less(waiter, entry)) {
315− link = &parent->rb_left;
316− } else {
317− link = &parent->rb_right;
318− leftmost = false;
319− }
320− }
321−
322− rb_link_node(&waiter->pi_tree_entry, parent, link);
323− rb_insert_color_cached(&waiter->pi_tree_entry, &task->pi_waiters, leftmost);
305+ rb_add_cached(&waiter->pi_tree_entry, &task->pi_waiters, __pi_waiter_less);
324306 }
325307
326308 static void