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
20namespace 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
31inline 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
36class MutexLock {
37 public:
38 explicit MutexLock(pthread_mutex_t &mutex) : mutex_(&mutex) {
39 pthread_mutex_lock(mutex: mutex_);
40 }
41 ~MutexLock() { pthread_mutex_unlock(mutex: mutex_); }
42
43 private:
44 pthread_mutex_t *mutex_ = nullptr;
45};
46
47} // namespace safestack
48
49#endif // SAFESTACK_UTIL_H
50

source code of compiler-rt/lib/safestack/safestack_util.h