| //===----------------------------------------------------------------------===// |
| // |
| // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. |
| // See https://llvm.org/LICENSE.txt for license information. |
| // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception |
| // |
| //===----------------------------------------------------------------------===// |
| |
| #ifndef ASAN_TESTING_H |
| #define ASAN_TESTING_H |
| |
| #include "test_macros.h" |
| #include <vector> |
| #include <string> |
| #include <memory> |
| #include <type_traits> |
| |
| #if TEST_HAS_FEATURE(address_sanitizer) |
| extern "C" int __sanitizer_verify_contiguous_container(const void* beg, const void* mid, const void* end); |
| |
| template <typename T, typename Alloc> |
| TEST_CONSTEXPR bool is_contiguous_container_asan_correct(const std::vector<T, Alloc>& c) { |
| if (TEST_IS_CONSTANT_EVALUATED) |
| return true; |
| if (std::is_same<Alloc, std::allocator<T> >::value && c.data() != NULL) |
| return __sanitizer_verify_contiguous_container(c.data(), c.data() + c.size(), c.data() + c.capacity()) != 0; |
| return true; |
| } |
| #else |
| template <typename T, typename Alloc> |
| TEST_CONSTEXPR bool is_contiguous_container_asan_correct(const std::vector<T, Alloc>&) { |
| return true; |
| } |
| #endif // TEST_HAS_FEATURE(address_sanitizer) |
| |
| #if TEST_HAS_FEATURE(address_sanitizer) |
| extern "C" int __sanitizer_verify_double_ended_contiguous_container( |
| const void* beg, const void* con_beg, const void* con_end, const void* end); |
| extern "C" bool __sanitizer_is_annotable(const void* address, const unsigned long size); |
| #include <deque> |
| |
| template <class T, class Alloc> |
| TEST_CONSTEXPR bool is_double_ended_contiguous_container_asan_correct(const std::deque<T, Alloc>& c) { |
| if (TEST_IS_CONSTANT_EVALUATED) |
| return true; |
| if (std::is_same<Alloc, std::allocator<T> >::value) |
| return c.__verify_asan_annotations(); |
| return true; |
| } |
| #else |
| # include <deque> |
| template <class T, class Alloc> |
| TEST_CONSTEXPR bool is_double_ended_contiguous_container_asan_correct(const std::deque<T, Alloc>&) { |
| return true; |
| } |
| #endif |
| |
| #if TEST_HAS_FEATURE(address_sanitizer) |
| template <typename S> |
| bool is_string_short(S const& s) { |
| // We do not have access to __is_long(), but we can check if strings |
| // buffer is inside strings memory. If strings memory contains its content, |
| // SSO is in use. To check it, we can just confirm that the beginning is in |
| // the string object memory block. |
| // &s - beginning of objects memory |
| // &s[0] - beginning of the buffer |
| // (&s+1) - end of objects memory |
| return (void*)std::addressof(s) <= (void*)std::addressof(s[0]) && |
| (void*)std::addressof(s[0]) < (void*)(std::addressof(s) + 1); |
| } |
| |
| template <typename ChrT, typename TraitsT, typename Alloc> |
| TEST_CONSTEXPR bool is_string_asan_correct(const std::basic_string<ChrT, TraitsT, Alloc>& c) { |
| if (TEST_IS_CONSTANT_EVALUATED) |
| return true; |
| |
| if (!is_string_short(c) || _LIBCPP_SHORT_STRING_ANNOTATIONS_ALLOWED) { |
| if (std::__asan_annotate_container_with_allocator<Alloc>::value) |
| return __sanitizer_verify_contiguous_container(c.data(), c.data() + c.size() + 1, c.data() + c.capacity() + 1) != |
| 0; |
| else |
| return __sanitizer_verify_contiguous_container( |
| c.data(), c.data() + c.capacity() + 1, c.data() + c.capacity() + 1) != 0; |
| } else { |
| return __sanitizer_verify_contiguous_container(std::addressof(c), std::addressof(c) + 1, std::addressof(c) + 1) != |
| 0; |
| } |
| } |
| #else |
| # include <string> |
| template <typename ChrT, typename TraitsT, typename Alloc> |
| TEST_CONSTEXPR bool is_string_asan_correct(const std::basic_string<ChrT, TraitsT, Alloc>&) { |
| return true; |
| } |
| #endif // TEST_HAS_FEATURE(address_sanitizer) |
| #endif // ASAN_TESTING_H |