perf probe: Fix memory leak when synthesizing SDT probes

[linux-2.6-microblaze.git] / lib / sbitmap.c

diff --git a/lib/sbitmap.c b/lib/sbitmap.c
index 4fd8770..d693d92 100644 (file)
--- a/lib/sbitmap.c
+++ b/lib/sbitmap.c
@@ -14,7 +14,7 @@
  */
 static inline bool sbitmap_deferred_clear(struct sbitmap_word *map)
 {
-       unsigned long mask, val;
+       unsigned long mask;
 
        if (!READ_ONCE(map->cleared))
                return false;
@@ -27,10 +27,8 @@ static inline bool sbitmap_deferred_clear(struct sbitmap_word *map)
        /*
         * Now clear the masked bits in our free word
         */
-       do {
-               val = map->word;
-       } while (cmpxchg(&map->word, val, val & ~mask) != val);
-
+       atomic_long_andnot(mask, (atomic_long_t *)&map->word);
+       BUILD_BUG_ON(sizeof(atomic_long_t) != sizeof(map->word));
        return true;
 }
 
@@ -99,9 +97,11 @@ EXPORT_SYMBOL_GPL(sbitmap_resize);
 static int __sbitmap_get_word(unsigned long *word, unsigned long depth,
                              unsigned int hint, bool wrap)
 {
-       unsigned int orig_hint = hint;
        int nr;
 
+       /* don't wrap if starting from 0 */
+       wrap = wrap && hint;
+
        while (1) {
                nr = find_next_zero_bit(word, depth, hint);
                if (unlikely(nr >= depth)) {
@@ -110,8 +110,8 @@ static int __sbitmap_get_word(unsigned long *word, unsigned long depth,
                         * offset to 0 in a failure case, so start from 0 to
                         * exhaust the map.
                         */
-                       if (orig_hint && hint && wrap) {
-                               hint = orig_hint = 0;
+                       if (hint && wrap) {
+                               hint = 0;
                                continue;
                        }
                        return -1;