| // Copyright 2017 The Crashpad Authors |
| // |
| // Licensed under the Apache License, Version 2.0 (the "License"); |
| // you may not use this file except in compliance with the License. |
| // You may obtain a copy of the License at |
| // |
| // http://www.apache.org/licenses/LICENSE-2.0 |
| // |
| // Unless required by applicable law or agreed to in writing, software |
| // distributed under the License is distributed on an "AS IS" BASIS, |
| // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| // See the License for the specific language governing permissions and |
| // limitations under the License. |
| |
| #include "client/annotation.h" |
| |
| #include <algorithm> |
| #include <iterator> |
| #include <string> |
| #include <type_traits> |
| #include <vector> |
| |
| #include "base/rand_util.h" |
| #include "client/crashpad_info.h" |
| #include "gtest/gtest.h" |
| #include "util/misc/clock.h" |
| #include "util/thread/thread.h" |
| |
| namespace crashpad { |
| namespace test { |
| namespace { |
| |
| #if (__cplusplus >= 202002L) |
| template <typename Iterator> |
| requires std::input_iterator<Iterator> |
| void VerifyIsInputIterator(Iterator) {} |
| #else |
| template <typename Iterator> |
| struct IsLegacyIteratorImpl { |
| static constexpr bool value = |
| std::is_copy_constructible_v<Iterator> && |
| std::is_copy_assignable_v<Iterator> && std::is_destructible_v<Iterator> && |
| std::is_swappable_v<Iterator> && |
| // check that std::iterator_traits has the necessary types (check only one |
| // needed as std::iterator is required to define only if all are defined) |
| !std::is_same_v<typename std::iterator_traits<Iterator>::reference, |
| void> && |
| std::is_same_v<decltype(++std::declval<Iterator>()), Iterator&> && |
| !std::is_same_v<decltype(*std::declval<Iterator>()), void>; |
| }; |
| |
| template <typename Iterator> |
| struct IsLegacyInputIteratorImpl { |
| static constexpr bool value = |
| IsLegacyIteratorImpl<Iterator>::value && |
| std::is_base_of_v< |
| std::input_iterator_tag, |
| typename std::iterator_traits<Iterator>::iterator_category> && |
| std::is_convertible_v<decltype(std::declval<Iterator>() != |
| std::declval<Iterator>()), |
| bool> && |
| std::is_convertible_v<decltype(std::declval<Iterator>() == |
| std::declval<Iterator>()), |
| bool> && |
| std::is_same_v<decltype(*std::declval<Iterator>()), |
| typename std::iterator_traits<Iterator>::reference> && |
| std::is_same_v<decltype(++std::declval<Iterator>()), Iterator&> && |
| std::is_same_v<decltype(std::declval<Iterator>()++), Iterator> && |
| std::is_same_v<decltype(*(++std::declval<Iterator>())), |
| typename std::iterator_traits<Iterator>::reference>; |
| }; |
| |
| template <typename Iterator> |
| void VerifyIsInputIterator(Iterator) { |
| static_assert(IsLegacyInputIteratorImpl<Iterator>::value); |
| } |
| #endif |
| |
| TEST(AnnotationListStatic, Register) { |
| ASSERT_FALSE(AnnotationList::Get()); |
| EXPECT_TRUE(AnnotationList::Register()); |
| EXPECT_TRUE(AnnotationList::Get()); |
| EXPECT_EQ(AnnotationList::Get(), AnnotationList::Register()); |
| |
| // This isn't expected usage of the AnnotationList API, but it is necessary |
| // for testing. |
| AnnotationList* list = AnnotationList::Get(); |
| CrashpadInfo::GetCrashpadInfo()->set_annotations_list(nullptr); |
| delete list; |
| |
| EXPECT_FALSE(AnnotationList::Get()); |
| } |
| |
| class AnnotationList : public testing::Test { |
| public: |
| void SetUp() override { |
| CrashpadInfo::GetCrashpadInfo()->set_annotations_list(&annotations_); |
| } |
| |
| void TearDown() override { |
| CrashpadInfo::GetCrashpadInfo()->set_annotations_list(nullptr); |
| } |
| |
| // NOTE: Annotations should be declared at file-scope, but in order to test |
| // them, they are declared as part of the test. These members are public so |
| // they are accessible from global helpers. |
| crashpad::StringAnnotation<8> one_{"First"}; |
| crashpad::StringAnnotation<256> two_{"Second"}; |
| crashpad::StringAnnotation<101> three_{"First"}; |
| |
| protected: |
| using AllAnnotations = std::vector<std::pair<std::string, std::string>>; |
| |
| AllAnnotations CollectAnnotations() { |
| AllAnnotations annotations; |
| |
| for (Annotation* curr : annotations_) { |
| if (!curr->is_set()) |
| continue; |
| std::string value(static_cast<const char*>(curr->value()), curr->size()); |
| annotations.push_back(std::make_pair(curr->name(), value)); |
| } |
| |
| return annotations; |
| } |
| |
| bool ContainsNameValue(const AllAnnotations& annotations, |
| const std::string& name, |
| const std::string& value) { |
| return std::find(annotations.begin(), |
| annotations.end(), |
| std::make_pair(name, value)) != annotations.end(); |
| } |
| |
| crashpad::AnnotationList annotations_; |
| }; |
| |
| TEST_F(AnnotationList, SetAndClear) { |
| one_.Set("this is a value longer than 8 bytes"); |
| AllAnnotations annotations = CollectAnnotations(); |
| |
| EXPECT_EQ(1u, annotations.size()); |
| EXPECT_TRUE(ContainsNameValue(annotations, "First", "this is ")); |
| |
| one_.Clear(); |
| |
| EXPECT_EQ(0u, CollectAnnotations().size()); |
| |
| one_.Set("short"); |
| two_.Set(std::string(500, 'A').data()); |
| |
| annotations = CollectAnnotations(); |
| EXPECT_EQ(2u, annotations.size()); |
| |
| EXPECT_EQ(5u, one_.size()); |
| EXPECT_EQ(256u, two_.size()); |
| |
| EXPECT_TRUE(ContainsNameValue(annotations, "First", "short")); |
| EXPECT_TRUE(ContainsNameValue(annotations, "Second", std::string(256, 'A'))); |
| } |
| |
| TEST_F(AnnotationList, DuplicateKeys) { |
| ASSERT_EQ(0u, CollectAnnotations().size()); |
| |
| one_.Set("1"); |
| three_.Set("2"); |
| |
| AllAnnotations annotations = CollectAnnotations(); |
| EXPECT_EQ(2u, annotations.size()); |
| |
| EXPECT_TRUE(ContainsNameValue(annotations, "First", "1")); |
| EXPECT_TRUE(ContainsNameValue(annotations, "First", "2")); |
| |
| one_.Clear(); |
| |
| annotations = CollectAnnotations(); |
| EXPECT_EQ(1u, annotations.size()); |
| } |
| |
| TEST_F(AnnotationList, IteratorSingleAnnotation) { |
| ASSERT_EQ(annotations_.begin(), annotations_.end()); |
| ASSERT_EQ(annotations_.cbegin(), annotations_.cend()); |
| |
| one_.Set("1"); |
| |
| auto iterator = annotations_.begin(); |
| auto const_iterator = annotations_.cbegin(); |
| |
| ASSERT_NE(iterator, annotations_.end()); |
| ASSERT_NE(const_iterator, annotations_.cend()); |
| |
| EXPECT_EQ(*iterator, &one_); |
| EXPECT_EQ(*const_iterator, &one_); |
| |
| ++iterator; |
| ++const_iterator; |
| |
| EXPECT_EQ(iterator, annotations_.end()); |
| EXPECT_EQ(const_iterator, annotations_.cend()); |
| } |
| |
| TEST_F(AnnotationList, IteratorMultipleAnnotationsInserted) { |
| ASSERT_EQ(annotations_.begin(), annotations_.end()); |
| ASSERT_EQ(annotations_.cbegin(), annotations_.cend()); |
| |
| one_.Set("1"); |
| two_.Set("2"); |
| |
| // New annotations are inserted to the beginning of the list. Hence, |two_| |
| // must be the first annotation, followed by |one_|. |
| { |
| auto iterator = annotations_.begin(); |
| auto const_iterator = annotations_.cbegin(); |
| |
| ASSERT_NE(iterator, annotations_.end()); |
| ASSERT_NE(const_iterator, annotations_.cend()); |
| |
| EXPECT_EQ(*iterator, &two_); |
| EXPECT_EQ(*const_iterator, &two_); |
| |
| ++iterator; |
| ++const_iterator; |
| |
| ASSERT_NE(iterator, annotations_.end()); |
| ASSERT_NE(const_iterator, annotations_.cend()); |
| |
| EXPECT_EQ(*iterator, &one_); |
| EXPECT_EQ(*const_iterator, &one_); |
| |
| ++iterator; |
| ++const_iterator; |
| |
| EXPECT_EQ(iterator, annotations_.end()); |
| EXPECT_EQ(const_iterator, annotations_.cend()); |
| } |
| } |
| |
| TEST_F(AnnotationList, IteratorMultipleAnnotationsInsertedAndRemoved) { |
| ASSERT_EQ(annotations_.begin(), annotations_.end()); |
| ASSERT_EQ(annotations_.cbegin(), annotations_.cend()); |
| |
| one_.Set("1"); |
| two_.Set("2"); |
| one_.Clear(); |
| two_.Clear(); |
| |
| // Even after clearing, Annotations are still inserted in the list and |
| // reachable via the iterators. |
| auto iterator = annotations_.begin(); |
| auto const_iterator = annotations_.cbegin(); |
| |
| ASSERT_NE(iterator, annotations_.end()); |
| ASSERT_NE(const_iterator, annotations_.cend()); |
| |
| EXPECT_EQ(*iterator, &two_); |
| EXPECT_EQ(*const_iterator, &two_); |
| |
| ++iterator; |
| ++const_iterator; |
| |
| ASSERT_NE(iterator, annotations_.end()); |
| ASSERT_NE(const_iterator, annotations_.cend()); |
| |
| EXPECT_EQ(*iterator, &one_); |
| EXPECT_EQ(*const_iterator, &one_); |
| |
| ++iterator; |
| ++const_iterator; |
| |
| EXPECT_EQ(iterator, annotations_.end()); |
| EXPECT_EQ(const_iterator, annotations_.cend()); |
| } |
| |
| TEST_F(AnnotationList, IteratorIsInputIterator) { |
| one_.Set("1"); |
| two_.Set("2"); |
| |
| // Check explicitly that the iterators meet the interface of an input |
| // iterator. |
| VerifyIsInputIterator(annotations_.begin()); |
| VerifyIsInputIterator(annotations_.cbegin()); |
| VerifyIsInputIterator(annotations_.end()); |
| VerifyIsInputIterator(annotations_.cend()); |
| |
| // Additionally verify that std::distance accepts the iterators. It requires |
| // the iterators to comply to the input iterator interface. |
| EXPECT_EQ(std::distance(annotations_.begin(), annotations_.end()), 2); |
| EXPECT_EQ(std::distance(annotations_.cbegin(), annotations_.cend()), 2); |
| } |
| |
| class RaceThread : public Thread { |
| public: |
| explicit RaceThread(test::AnnotationList* test) : Thread(), test_(test) {} |
| |
| private: |
| void ThreadMain() override { |
| for (int i = 0; i <= 50; ++i) { |
| if (i % 2 == 0) { |
| test_->three_.Set("three"); |
| test_->two_.Clear(); |
| } else { |
| test_->three_.Clear(); |
| } |
| SleepNanoseconds(base::RandInt(1, 1000)); |
| } |
| } |
| |
| test::AnnotationList* test_; |
| }; |
| |
| TEST_F(AnnotationList, MultipleThreads) { |
| ASSERT_EQ(0u, CollectAnnotations().size()); |
| |
| RaceThread other_thread(this); |
| other_thread.Start(); |
| |
| for (int i = 0; i <= 50; ++i) { |
| if (i % 2 == 0) { |
| one_.Set("one"); |
| two_.Set("two"); |
| } else { |
| one_.Clear(); |
| } |
| SleepNanoseconds(base::RandInt(1, 1000)); |
| } |
| |
| other_thread.Join(); |
| |
| AllAnnotations annotations = CollectAnnotations(); |
| EXPECT_GE(annotations.size(), 2u); |
| EXPECT_LE(annotations.size(), 3u); |
| |
| EXPECT_TRUE(ContainsNameValue(annotations, "First", "one")); |
| EXPECT_TRUE(ContainsNameValue(annotations, "First", "three")); |
| |
| if (annotations.size() == 3) { |
| EXPECT_TRUE(ContainsNameValue(annotations, "Second", "two")); |
| } |
| } |
| |
| } // namespace |
| } // namespace test |
| } // namespace crashpad |