2 files changed, 12 insertions, 22 deletions

diff --git a/include/linux/llist.h b/include/linux/llist.h
index a5199f6d0e82..3e2b969d68f6 100644
--- a/include/linux/llist.h
+++ b/include/linux/llist.h
@@ -151,18 +151,13 @@ static inline struct llist_node *llist_next(struct llist_node *node)
  */
 static inline bool llist_add(struct llist_node *new, struct llist_head *head)
 {
-        struct llist_node *entry, *old_entry;
-
-        entry = head->first;
-        for (;;) {
-                old_entry = entry;
-                new->next = entry;
-                entry = cmpxchg(&head->first, old_entry, new);
-                if (entry == old_entry)
-                        break;
-        }
-
-        return old_entry == NULL;
+        struct llist_node *first;
+
+        do {
+                new->next = first = ACCESS_ONCE(head->first);
+        } while (cmpxchg(&head->first, first, new) != first);
+
+        return !first;
 }
 
 /**
diff --git a/lib/llist.c b/lib/llist.c
index 4a15115e90f8..4a70d120138c 100644
--- a/lib/llist.c
+++ b/lib/llist.c
@@ -39,18 +39,13 @@
 bool llist_add_batch(struct llist_node *new_first, struct llist_node *new_last,
                      struct llist_head *head)
 {
-        struct llist_node *entry, *old_entry;
+        struct llist_node *first;
 
-        entry = head->first;
-        for (;;) {
-                old_entry = entry;
-                new_last->next = entry;
-                entry = cmpxchg(&head->first, old_entry, new_first);
-                if (entry == old_entry)
-                        break;
-        }
+        do {
+                new_last->next = first = ACCESS_ONCE(head->first);
+        } while (cmpxchg(&head->first, first, new_first) != first);
 
-        return old_entry == NULL;
+        return !first;
 }
 EXPORT_SYMBOL_GPL(llist_add_batch);