| 1 | //===----------------------------------------------------------------------===// |
| 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 | // UNSUPPORTED: c++03, c++11, c++14, c++17 |
| 10 | |
| 11 | // If we have a copy-propagating cache, when we copy ZeroOnDestroy, we will get a |
| 12 | // dangling reference to the copied-from object. This test ensures that we do not |
| 13 | // propagate the cache on copy. |
| 14 | |
| 15 | #include <ranges> |
| 16 | |
| 17 | #include <cstddef> |
| 18 | #include <cstring> |
| 19 | |
| 20 | #include "test_macros.h" |
| 21 | #include "types.h" |
| 22 | |
| 23 | struct ZeroOnDestroy : std::ranges::view_base { |
| 24 | unsigned count = 0; |
| 25 | int buff[8] = {1, 2, 3, 4, 5, 6, 7, 8}; |
| 26 | |
| 27 | constexpr ForwardIter begin() { return ForwardIter(buff); } |
| 28 | constexpr ForwardIter begin() const { return ForwardIter(); } |
| 29 | constexpr ForwardIter end() { return ForwardIter(buff + 8); } |
| 30 | constexpr ForwardIter end() const { return ForwardIter(); } |
| 31 | |
| 32 | ZeroOnDestroy() = default; |
| 33 | ZeroOnDestroy(const ZeroOnDestroy&) = default; |
| 34 | ZeroOnDestroy& operator=(const ZeroOnDestroy&) = default; |
| 35 | ~ZeroOnDestroy() { |
| 36 | std::memset(s: buff, c: 0, n: sizeof(buff)); |
| 37 | } |
| 38 | |
| 39 | static auto dropFirstFour() { |
| 40 | ZeroOnDestroy zod; |
| 41 | std::ranges::drop_view dv(zod, 4); |
| 42 | // Make sure we call begin here so the next call to begin will |
| 43 | // use the cached iterator. |
| 44 | assert(*dv.begin() == 5); |
| 45 | // Intentionally invoke the copy ctor here. |
| 46 | return std::ranges::drop_view(dv); |
| 47 | } |
| 48 | }; |
| 49 | |
| 50 | int main(int, char**) { |
| 51 | auto noDanglingCache = ZeroOnDestroy::dropFirstFour(); |
| 52 | // If we use the cached version, it will reference the copied-from view. |
| 53 | // Worst case this is a segfault, best case it's an assertion fired. |
| 54 | assert(*noDanglingCache.begin() == 5); |
| 55 | |
| 56 | return 0; |
| 57 | } |
| 58 | |