reference, declaration → definition
definition → references, declarations, derived classes, virtual overrides
reference to multiple definitions → definitions
unreferenced

    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

//===-- mutex.h -------------------------------------------------*- C++ -*-===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//

#ifndef GWP_ASAN_MUTEX_H_
#define GWP_ASAN_MUTEX_H_

#ifdef __unix__
#include <pthread.h>
#else
#error "GWP-ASan is not supported on this platform."
#endif

namespace gwp_asan {
class Mutex {
public:
  constexpr Mutex() = default;
  ~Mutex() = default;
  Mutex(const Mutex &) = delete;
  Mutex &operator=(const Mutex &) = delete;
  // Lock the mutex.
  void lock();
  // Nonblocking trylock of the mutex. Returns true if the lock was acquired.
  bool tryLock();
  // Unlock the mutex.
  void unlock();

private:
#ifdef __unix__
  pthread_mutex_t Mu = PTHREAD_MUTEX_INITIALIZER;
#endif // defined(__unix__)
};

class ScopedLock {
public:
  explicit ScopedLock(Mutex &Mx) : Mu(Mx) { Mu.lock(); }
  ~ScopedLock() { Mu.unlock(); }
  ScopedLock(const ScopedLock &) = delete;
  ScopedLock &operator=(const ScopedLock &) = delete;

private:
  Mutex &Mu;
};
} // namespace gwp_asan

#endif // GWP_ASAN_MUTEX_H_