Branch data     Line data    Source code

#       1                 :            : // Copyright (c) 2015-2021 The Bitcoin Core developers
#       2                 :            : // Distributed under the MIT software license, see the accompanying
#       3                 :            : // file COPYING or http://www.opensource.org/licenses/mit-license.php.
#       4                 :            : 
#       5                 :            : #include <sync.h>
#       6                 :            : #include <test/util/setup_common.h>
#       7                 :            : 
#       8                 :            : #include <boost/test/unit_test.hpp>
#       9                 :            : 
#      10                 :            : #include <stdexcept>
#      11                 :            : 
#      12                 :            : BOOST_AUTO_TEST_SUITE(reverselock_tests)
#      13                 :            : 
#      14                 :            : BOOST_AUTO_TEST_CASE(reverselock_basics)
#      15                 :          2 : {
#      16                 :          2 :     Mutex mutex;
#      17                 :          2 :     WAIT_LOCK(mutex, lock);
#      18                 :            : 
#      19                 :          2 :     BOOST_CHECK(lock.owns_lock());
#      20                 :          2 :     {
#      21                 :          2 :         REVERSE_LOCK(lock);
#      22                 :          2 :         BOOST_CHECK(!lock.owns_lock());
#      23                 :          2 :     }
#      24                 :          2 :     BOOST_CHECK(lock.owns_lock());
#      25                 :          2 : }
#      26                 :            : 
#      27                 :            : BOOST_AUTO_TEST_CASE(reverselock_multiple)
#      28                 :          2 : {
#      29                 :          2 :     Mutex mutex2;
#      30                 :          2 :     Mutex mutex;
#      31                 :          2 :     WAIT_LOCK(mutex2, lock2);
#      32                 :          2 :     WAIT_LOCK(mutex, lock);
#      33                 :            : 
#      34                 :            :     // Make sure undoing two locks succeeds
#      35                 :          2 :     {
#      36                 :          2 :         REVERSE_LOCK(lock);
#      37                 :          2 :         BOOST_CHECK(!lock.owns_lock());
#      38                 :          2 :         REVERSE_LOCK(lock2);
#      39                 :          2 :         BOOST_CHECK(!lock2.owns_lock());
#      40                 :          2 :     }
#      41                 :          2 :     BOOST_CHECK(lock.owns_lock());
#      42                 :          2 :     BOOST_CHECK(lock2.owns_lock());
#      43                 :          2 : }
#      44                 :            : 
#      45                 :            : BOOST_AUTO_TEST_CASE(reverselock_errors)
#      46                 :          2 : {
#      47                 :          2 :     Mutex mutex2;
#      48                 :          2 :     Mutex mutex;
#      49                 :          2 :     WAIT_LOCK(mutex2, lock2);
#      50                 :          2 :     WAIT_LOCK(mutex, lock);
#      51                 :            : 
#      52                 :          2 : #ifdef DEBUG_LOCKORDER
#      53                 :          2 :     bool prev = g_debug_lockorder_abort;
#      54                 :          2 :     g_debug_lockorder_abort = false;
#      55                 :            : 
#      56                 :            :     // Make sure trying to reverse lock a previous lock fails
#      57                 :          2 :     BOOST_CHECK_EXCEPTION(REVERSE_LOCK(lock2), std::logic_error, HasReason("lock2 was not most recent critical section locked"));
#      58                 :          2 :     BOOST_CHECK(lock2.owns_lock());
#      59                 :            : 
#      60                 :          2 :     g_debug_lockorder_abort = prev;
#      61                 :          2 : #endif
#      62                 :            : 
#      63                 :            :     // Make sure trying to reverse lock an unlocked lock fails
#      64                 :          2 :     lock.unlock();
#      65                 :            : 
#      66                 :          2 :     BOOST_CHECK(!lock.owns_lock());
#      67                 :            : 
#      68                 :          2 :     bool failed = false;
#      69                 :          2 :     try {
#      70                 :          2 :         REVERSE_LOCK(lock);
#      71                 :          2 :     } catch(...) {
#      72                 :          2 :         failed = true;
#      73                 :          2 :     }
#      74                 :            : 
#      75                 :          2 :     BOOST_CHECK(failed);
#      76                 :          2 :     BOOST_CHECK(!lock.owns_lock());
#      77                 :            : 
#      78                 :            :     // Locking the original lock after it has been taken by a reverse lock
#      79                 :            :     // makes no sense. Ensure that the original lock no longer owns the lock
#      80                 :            :     // after giving it to a reverse one.
#      81                 :            : 
#      82                 :          2 :     lock.lock();
#      83                 :          2 :     BOOST_CHECK(lock.owns_lock());
#      84                 :          2 :     {
#      85                 :          2 :         REVERSE_LOCK(lock);
#      86                 :          2 :         BOOST_CHECK(!lock.owns_lock());
#      87                 :          2 :     }
#      88                 :            : 
#      89                 :          2 :     BOOST_CHECK(failed);
#      90                 :          2 :     BOOST_CHECK(lock.owns_lock());
#      91                 :          2 : }
#      92                 :            : 
#      93                 :            : BOOST_AUTO_TEST_SUITE_END()