031384d8521abd48fe36e9da2e35cbb74437cb64
[WebKit.git] / Source / WTF / wtf / RecursiveLockAdapter.h
1 /*
2  * Copyright (C) 2016 Apple Inc.  All rights reserved.
3  *
4  * Redistribution and use in source and binary forms, with or without
5  * modification, are permitted provided that the following conditions
6  * are met:
7  * 1. Redistributions of source code must retain the above copyright
8  *    notice, this list of conditions and the following disclaimer.
9  * 2. Redistributions in binary form must reproduce the above copyright
10  *    notice, this list of conditions and the following disclaimer in the
11  *    documentation and/or other materials provided with the distribution.
12  *
13  * THIS SOFTWARE IS PROVIDED BY APPLE INC. ``AS IS'' AND ANY
14  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
15  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
16  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL APPLE INC. OR
17  * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
18  * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
19  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
20  * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
21  * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
22  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
23  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 
24  */
25
26 #pragma once
27
28 #include <wtf/Threading.h>
29
30 namespace WTF {
31
32 template<typename LockType>
33 class RecursiveLockAdapter {
34 public:
35     RecursiveLockAdapter() = default;
36
37     void lock()
38     {
39         Thread& me = Thread::current();
40         if (&me == m_owner) {
41             m_recursionCount++;
42             return;
43         }
44         
45         m_lock.lock();
46         ASSERT(!m_owner);
47         ASSERT(!m_recursionCount);
48         m_owner = &me;
49         m_recursionCount = 1;
50     }
51     
52     void unlock()
53     {
54         if (--m_recursionCount)
55             return;
56         m_owner = nullptr;
57         m_lock.unlock();
58     }
59     
60     bool tryLock()
61     {
62         Thread& me = Thread::current();
63         if (&me == m_owner) {
64             m_recursionCount++;
65             return true;
66         }
67         
68         if (!m_lock.tryLock())
69             return false;
70         
71         ASSERT(!m_owner);
72         ASSERT(!m_recursionCount);
73         m_owner = &me;
74         m_recursionCount = 1;
75         return true;
76     }
77     
78     bool isLocked() const
79     {
80         return m_lock.isLocked();
81     }
82     
83 private:
84     Thread* m_owner { nullptr }; // Use Thread* instead of RefPtr<Thread> since m_owner thread is always alive while m_onwer is set.
85     unsigned m_recursionCount { 0 };
86     LockType m_lock;
87 };
88
89 } // namespace WTF
90
91