1*0fca6ea1SDimitry Andric //===--- rtsan_test_utilities.h - Realtime Sanitizer ------------*- C++ -*-===// 2*0fca6ea1SDimitry Andric // 3*0fca6ea1SDimitry Andric // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. 4*0fca6ea1SDimitry Andric // See https://llvm.org/LICENSE.txt for license information. 5*0fca6ea1SDimitry Andric // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception 6*0fca6ea1SDimitry Andric // 7*0fca6ea1SDimitry Andric //===----------------------------------------------------------------------===// 8*0fca6ea1SDimitry Andric // 9*0fca6ea1SDimitry Andric //===----------------------------------------------------------------------===// 10*0fca6ea1SDimitry Andric 11*0fca6ea1SDimitry Andric #pragma once 12*0fca6ea1SDimitry Andric 13*0fca6ea1SDimitry Andric #include "rtsan.h" 14*0fca6ea1SDimitry Andric #include "gmock/gmock.h" 15*0fca6ea1SDimitry Andric #include <string> 16*0fca6ea1SDimitry Andric 17*0fca6ea1SDimitry Andric namespace rtsan_testing { 18*0fca6ea1SDimitry Andric 19*0fca6ea1SDimitry Andric template <typename Function> void RealtimeInvoke(Function &&Func) { 20*0fca6ea1SDimitry Andric __rtsan_realtime_enter(); 21*0fca6ea1SDimitry Andric std::forward<Function>(Func)(); 22*0fca6ea1SDimitry Andric __rtsan_realtime_exit(); 23*0fca6ea1SDimitry Andric } 24*0fca6ea1SDimitry Andric 25*0fca6ea1SDimitry Andric template <typename Function> 26*0fca6ea1SDimitry Andric void ExpectRealtimeDeath(Function &&Func, 27*0fca6ea1SDimitry Andric const char *intercepted_method_name = nullptr) { 28*0fca6ea1SDimitry Andric 29*0fca6ea1SDimitry Andric using namespace testing; 30*0fca6ea1SDimitry Andric 31*0fca6ea1SDimitry Andric auto GetExpectedErrorSubstring = [&]() -> std::string { 32*0fca6ea1SDimitry Andric return intercepted_method_name != nullptr 33*0fca6ea1SDimitry Andric ? "Real-time violation: intercepted call to real-time unsafe " 34*0fca6ea1SDimitry Andric "function `" + 35*0fca6ea1SDimitry Andric std::string(intercepted_method_name) + "`" 36*0fca6ea1SDimitry Andric : ""; 37*0fca6ea1SDimitry Andric }; 38*0fca6ea1SDimitry Andric 39*0fca6ea1SDimitry Andric EXPECT_EXIT(RealtimeInvoke(std::forward<Function>(Func)), 40*0fca6ea1SDimitry Andric ExitedWithCode(EXIT_FAILURE), GetExpectedErrorSubstring()); 41*0fca6ea1SDimitry Andric } 42*0fca6ea1SDimitry Andric 43*0fca6ea1SDimitry Andric template <typename Function> void ExpectNonRealtimeSurvival(Function &&Func) { 44*0fca6ea1SDimitry Andric std::forward<Function>(Func)(); 45*0fca6ea1SDimitry Andric } 46*0fca6ea1SDimitry Andric 47*0fca6ea1SDimitry Andric } // namespace rtsan_testing 48