Branch data     Line data    Source code

#       1                 :            : // Copyright (c) 2015-2020 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                 :            : BOOST_FIXTURE_TEST_SUITE(reverselock_tests, BasicTestingSetup)
#      11                 :            : 
#      12                 :            : BOOST_AUTO_TEST_CASE(reverselock_basics)
#      13                 :          2 : {
#      14                 :          2 :     Mutex mutex;
#      15                 :          2 :     WAIT_LOCK(mutex, lock);
#      16                 :            : 
#      17                 :          2 :     BOOST_CHECK(lock.owns_lock());
#      18                 :          2 :     {
#      19                 :          2 :         REVERSE_LOCK(lock);
#      20                 :          2 :         BOOST_CHECK(!lock.owns_lock());
#      21                 :          2 :     }
#      22                 :          2 :     BOOST_CHECK(lock.owns_lock());
#      23                 :          2 : }
#      24                 :            : 
#      25                 :            : BOOST_AUTO_TEST_CASE(reverselock_multiple)
#      26                 :          2 : {
#      27                 :          2 :     Mutex mutex2;
#      28                 :          2 :     Mutex mutex;
#      29                 :          2 :     WAIT_LOCK(mutex2, lock2);
#      30                 :          2 :     WAIT_LOCK(mutex, lock);
#      31                 :            : 
#      32                 :            :     // Make sure undoing two locks succeeds
#      33                 :          2 :     {
#      34                 :          2 :         REVERSE_LOCK(lock);
#      35                 :          2 :         BOOST_CHECK(!lock.owns_lock());
#      36                 :          2 :         REVERSE_LOCK(lock2);
#      37                 :          2 :         BOOST_CHECK(!lock2.owns_lock());
#      38                 :          2 :     }
#      39                 :          2 :     BOOST_CHECK(lock.owns_lock());
#      40                 :          2 :     BOOST_CHECK(lock2.owns_lock());
#      41                 :          2 : }
#      42                 :            : 
#      43                 :            : BOOST_AUTO_TEST_CASE(reverselock_errors)
#      44                 :          2 : {
#      45                 :          2 :     Mutex mutex2;
#      46                 :          2 :     Mutex mutex;
#      47                 :          2 :     WAIT_LOCK(mutex2, lock2);
#      48                 :          2 :     WAIT_LOCK(mutex, lock);
#      49                 :            : 
#      50                 :          2 : #ifdef DEBUG_LOCKORDER
#      51                 :          2 :     bool prev = g_debug_lockorder_abort;
#      52                 :          2 :     g_debug_lockorder_abort = false;
#      53                 :            : 
#      54                 :            :     // Make sure trying to reverse lock a previous lock fails
#      55                 :          2 :     BOOST_CHECK_EXCEPTION(REVERSE_LOCK(lock2), std::logic_error, HasReason("lock2 was not most recent critical section locked"));
#      56                 :          2 :     BOOST_CHECK(lock2.owns_lock());
#      57                 :            : 
#      58                 :          2 :     g_debug_lockorder_abort = prev;
#      59                 :          2 : #endif
#      60                 :            : 
#      61                 :            :     // Make sure trying to reverse lock an unlocked lock fails
#      62                 :          2 :     lock.unlock();
#      63                 :            : 
#      64                 :          2 :     BOOST_CHECK(!lock.owns_lock());
#      65                 :            : 
#      66                 :          2 :     bool failed = false;
#      67                 :          2 :     try {
#      68                 :          2 :         REVERSE_LOCK(lock);
#      69                 :          2 :     } catch(...) {
#      70                 :          2 :         failed = true;
#      71                 :          2 :     }
#      72                 :            : 
#      73                 :          2 :     BOOST_CHECK(failed);
#      74                 :          2 :     BOOST_CHECK(!lock.owns_lock());
#      75                 :            : 
#      76                 :            :     // Locking the original lock after it has been taken by a reverse lock
#      77                 :            :     // makes no sense. Ensure that the original lock no longer owns the lock
#      78                 :            :     // after giving it to a reverse one.
#      79                 :            : 
#      80                 :          2 :     lock.lock();
#      81                 :          2 :     BOOST_CHECK(lock.owns_lock());
#      82                 :          2 :     {
#      83                 :          2 :         REVERSE_LOCK(lock);
#      84                 :          2 :         BOOST_CHECK(!lock.owns_lock());
#      85                 :          2 :     }
#      86                 :            : 
#      87                 :          2 :     BOOST_CHECK(failed);
#      88                 :          2 :     BOOST_CHECK(lock.owns_lock());
#      89                 :          2 : }
#      90                 :            : 
#      91                 :            : BOOST_AUTO_TEST_SUITE_END()