Blame view

RIOT/sys/cpp11-compat/include/riot/mutex.hpp 5.08 KB
fb11e647   vrobic   reseau statique a...
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
  /*
   * Copyright (C) 2015 Hamburg University of Applied Sciences (HAW)
   *
   * This file is subject to the terms and conditions of the GNU Lesser
   * General Public License v2.1. See the file LICENSE in the top level
   * directory for more details.
   */
  
  /**
   * @ingroup cpp11-compat
   * @{
   *
   * @file
   * @brief   C++11 mutex drop in replacement
   * @see     <a href="http://en.cppreference.com/w/cpp/thread/mutex">
   *            std::mutex, std::lock_guard and std::unique_lock
   *          </a>
   *
   * @author  Raphael Hiesgen <raphael.hiesgen (at) haw-hamburg.de>
   *
   * @}
   */
  
  #ifndef RIOT_MUTEX_HPP
  #define RIOT_MUTEX_HPP
  
  #include "mutex.h"
  
  #include <utility>
  #include <stdexcept>
  #include <system_error>
  
  namespace riot {
  
  /**
   * @brief C++11 complient implementation of mutex, uses the time point
   *              implemented in our chrono replacement instead of the specified
   *              one
   * @see   <a href="http://en.cppreference.com/w/cpp/thread/mutex">
   *          std::mutex
   *        </a>
   */
  class mutex {
   public:
    using native_handle_type = mutex_t*;
  
    inline constexpr mutex() noexcept : m_mtx{{0}} {}
    ~mutex();
  
    void lock();
    bool try_lock() noexcept;
    void unlock() noexcept;
  
    inline native_handle_type native_handle() { return &m_mtx; }
  
   private:
    mutex(const mutex&);
    mutex& operator=(const mutex&);
  
    mutex_t m_mtx;
  };
  
  struct defer_lock_t {};
  struct try_to_lock_t {};
  struct adopt_lock_t {};
  
  constexpr defer_lock_t defer_lock = defer_lock_t();
  constexpr try_to_lock_t try_to_lock = try_to_lock_t();
  constexpr adopt_lock_t adopt_lock = adopt_lock_t();
  
  /**
   * @brief C++11 complient implementation of unique lock
   * @see   <a href="http://en.cppreference.com/w/cpp/thread/lock_guard">
   *          std::lock_guard
   *        </a>
   */
  template <class Mutex>
  class lock_guard {
  
   public:
    using mutex_type = Mutex;
  
    inline explicit lock_guard(mutex_type& mtx) : m_mtx(mtx) { m_mtx.lock(); }
    inline lock_guard(mutex_type& mtx, adopt_lock_t) : m_mtx{mtx} {}
    inline ~lock_guard() { m_mtx.unlock(); }
  
   private:
    mutex_type& m_mtx;
  };
  
  /**
   * @brief C++11 complient implementation of unique lock
   * @see   <a href="http://en.cppreference.com/w/cpp/thread/unique_lock">
   *          std::unique_lock
   *        </a>
   */
  template <class Mutex>
  class unique_lock {
  
   public:
    using mutex_type = Mutex;
  
    inline unique_lock() noexcept : m_mtx{nullptr}, m_owns{false} {}
    inline explicit unique_lock(mutex_type& mtx) : m_mtx{&mtx}, m_owns{true} {
      m_mtx->lock();
    }
    inline unique_lock(mutex_type& mtx, defer_lock_t) noexcept : m_mtx{&mtx},
                                                                 m_owns{false} {}
    inline unique_lock(mutex_type& mtx, try_to_lock_t)
        : m_mtx{&mtx}, m_owns{mtx.try_lock()} {}
    inline unique_lock(mutex_type& mtx, adopt_lock_t)
        : m_mtx{&mtx}, m_owns{true} {}
    inline ~unique_lock() {
      if (m_owns) {
        m_mtx->unlock();
      }
    }
    inline unique_lock(unique_lock&& lock) noexcept : m_mtx{lock.m_mtx},
                                                      m_owns{lock.m_owns} {
      lock.m_mtx = nullptr;
      lock.m_owns = false;
    }
    inline unique_lock& operator=(unique_lock&& lock) noexcept {
      if (m_owns) {
        m_mtx->unlock();
      }
      m_mtx = lock.m_mtx;
      m_owns = lock.m_owns;
      lock.m_mtx = nullptr;
      lock.m_owns = false;
      return *this;
    }
  
    void lock();
    bool try_lock();
  
    void unlock();
  
    inline void swap(unique_lock& lock) noexcept {
      std::swap(m_mtx, lock.m_mtx);
      std::swap(m_owns, lock.m_owns);
    }
  
    inline mutex_type* release() noexcept {
      mutex_type* mtx = m_mtx;
      m_mtx = nullptr;
      m_owns = false;
      return mtx;
    }
  
    inline bool owns_lock() const noexcept { return m_owns; }
    inline explicit operator bool() const noexcept { return m_owns; }
    inline mutex_type* mutex() const noexcept { return m_mtx; }
  
   private:
    unique_lock(unique_lock const&);
    unique_lock& operator=(unique_lock const&);
  
    mutex_type* m_mtx;
    bool m_owns;
  };
  
  template <class Mutex>
  void unique_lock<Mutex>::lock() {
    if (m_mtx == nullptr) {
      throw std::system_error(
        std::make_error_code(std::errc::operation_not_permitted),
        "References null mutex.");
    }
    if (m_owns) {
      throw std::system_error(
        std::make_error_code(std::errc::resource_deadlock_would_occur),
        "Already locked.");
    }
    m_mtx->lock();
    m_owns = true;
  }
  
  template <class Mutex>
  bool unique_lock<Mutex>::try_lock() {
    if (m_mtx == nullptr) {
      throw std::system_error(
        std::make_error_code(std::errc::operation_not_permitted),
        "References null mutex.");
    }
    if (m_owns) {
      throw std::system_error(
        std::make_error_code(std::errc::resource_deadlock_would_occur),
        "Already locked.");
    }
    m_owns = m_mtx->try_lock();
    return m_owns;
  }
  
  template <class Mutex>
  void unique_lock<Mutex>::unlock() {
    if (!m_owns) {
      throw std::system_error(
        std::make_error_code(std::errc::operation_not_permitted),
        "Mutex not locked.");
    }
    m_mtx->unlock();
    m_owns = false;
  }
  
  template <class Mutex>
  inline void swap(unique_lock<Mutex>& lhs, unique_lock<Mutex>& rhs) noexcept {
    lhs.swap(rhs);
  }
  
  } // namespace riot
  
  #endif // RIOT_MUTEX_HPP