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