54 lines · c
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-exception6//7//===----------------------------------------------------------------------===//8//9// This file contains utility code for SafeStack implementation.10//11//===----------------------------------------------------------------------===//12 13#ifndef SAFESTACK_UTIL_H14#define SAFESTACK_UTIL_H15 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 36inline constexpr bool IsAligned(size_t a, size_t alignment) {37 return (a & (alignment - 1)) == 0;38}39 40class MutexLock {41 public:42 explicit MutexLock(pthread_mutex_t &mutex) : mutex_(&mutex) {43 pthread_mutex_lock(mutex_);44 }45 ~MutexLock() { pthread_mutex_unlock(mutex_); }46 47 private:48 pthread_mutex_t *mutex_ = nullptr;49};50 51} // namespace safestack52 53#endif // SAFESTACK_UTIL_H54