return ret;
 }
 
+static int reacquire_held_locks(struct task_struct *curr, unsigned int depth,
+                             int idx)
+{
+       struct held_lock *hlock;
+
+       for (hlock = curr->held_locks + idx; idx < depth; idx++, hlock++) {
+               if (!__lock_acquire(hlock->instance,
+                                   hlock_class(hlock)->subclass,
+                                   hlock->trylock,
+                                   hlock->read, hlock->check,
+                                   hlock->hardirqs_off,
+                                   hlock->nest_lock, hlock->acquire_ip,
+                                   hlock->references, hlock->pin_count))
+                       return 1;
+       }
+       return 0;
+}
+
 static int
 __lock_set_class(struct lockdep_map *lock, const char *name,
                 struct lock_class_key *key, unsigned int subclass,
        curr->lockdep_depth = i;
        curr->curr_chain_key = hlock->prev_chain_key;
 
-       for (; i < depth; i++) {
-               hlock = curr->held_locks + i;
-               if (!__lock_acquire(hlock->instance,
-                       hlock_class(hlock)->subclass, hlock->trylock,
-                               hlock->read, hlock->check, hlock->hardirqs_off,
-                               hlock->nest_lock, hlock->acquire_ip,
-                               hlock->references, hlock->pin_count))
-                       return 0;
-       }
+       if (reacquire_held_locks(curr, depth, i))
+               return 0;
 
        /*
         * I took it apart and put it back together again, except now I have
        curr->lockdep_depth = i;
        curr->curr_chain_key = hlock->prev_chain_key;
 
-       for (i++; i < depth; i++) {
-               hlock = curr->held_locks + i;
-               if (!__lock_acquire(hlock->instance,
-                       hlock_class(hlock)->subclass, hlock->trylock,
-                               hlock->read, hlock->check, hlock->hardirqs_off,
-                               hlock->nest_lock, hlock->acquire_ip,
-                               hlock->references, hlock->pin_count))
-                       return 0;
-       }
+       if (reacquire_held_locks(curr, depth, i + 1))
+               return 0;
 
        /*
         * We had N bottles of beer on the wall, we drank one, but now