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
50