compiler-rt/lib/safestack/safestack_util.h

   1 //===-- safestack_util.h --------------------------------------------------===//
   2 //
   3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
   4 // See https://llvm.org/LICENSE.txt for license information.
   5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
   6 //
   7 //===----------------------------------------------------------------------===//
   8 //
   9 // This file contains utility code for SafeStack implementation.
  10 //
  11 //===----------------------------------------------------------------------===//
  12
  13 #ifndef SAFESTACK_UTIL_H
  14 #define SAFESTACK_UTIL_H
  15
  16 #include <pthread.h>
  17 #include <stdio.h>
  18 #include <stdlib.h>
  19
  20 namespace safestack {
  21
  22 #define SFS_CHECK(a)                                                  \
  23   do {                                                                \
  24     if (!(a)) {                                                       \
  25       fprintf(stderr, "safestack CHECK failed: %s:%d %s\n", __FILE__, \
  26               __LINE__, #a);                                          \
  27       abort();                                                        \
  28     };                                                                \
  29   } while (false)
  30
  31 inline size_t RoundUpTo(size_t size, size_t boundary) {
  32   SFS_CHECK((boundary & (boundary - 1)) == 0);
  33   return (size + boundary - 1) & ~(boundary - 1);
  34 }
  35
  36 class MutexLock {
  37  public:
  38   explicit MutexLock(pthread_mutex_t &mutex) : mutex_(&mutex) {
  39     pthread_mutex_lock(mutex_);
  40   }
  41   ~MutexLock() { pthread_mutex_unlock(mutex_); }
  42
  43  private:
  44   pthread_mutex_t *mutex_ = nullptr;
  45 };
  46
  47 }  // namespace safestack
  48
  49 #endif  // SAFESTACK_UTIL_H