Merge tag 'linux-kselftest-kunit-fixes-5.14-rc1' of git://git.kernel.org/pub/scm...

[sfrench/cifs-2.6.git] / include / kunit / test.h

diff --git a/include/kunit/test.h b/include/kunit/test.h
index 35b0aed9b739caa40bff0a9f82398814fbfe45c9..24b40e5c160b2b43bdbe58bfedfc58ba2855aeff 100644 (file)
--- a/include/kunit/test.h
+++ b/include/kunit/test.h
@@ -540,8 +540,9 @@ kunit_find_resource(struct kunit *test,
                    void *match_data)
 {
        struct kunit_resource *res, *found = NULL;
+       unsigned long flags;
 
-       spin_lock(&test->lock);
+       spin_lock_irqsave(&test->lock, flags);
 
        list_for_each_entry_reverse(res, &test->resources, node) {
                if (match(test, res, (void *)match_data)) {
@@ -551,7 +552,7 @@ kunit_find_resource(struct kunit *test,
                }
        }
 
-       spin_unlock(&test->lock);
+       spin_unlock_irqrestore(&test->lock, flags);
 
        return found;
 }