1 //===----------------------------------------------------------------------===//
2 //
3 //                     The LLVM Compiler Infrastructure
4 //
5 // This file is dual licensed under the MIT and the University of Illinois Open
6 // Source Licenses. See LICENSE.TXT for details.
7 //
8 //===----------------------------------------------------------------------===//
9 
10 // <vector>
11 
12 // void push_back(const value_type& x);
13 
14 #include <vector>
15 #include <cassert>
16 
17 #include "asan_testing.h"
18 
19 // Flag that makes the copy constructor for CMyClass throw an exception
20 static bool gCopyConstructorShouldThow = false;
21 
22 
23 class CMyClass {
24     public: CMyClass(int tag);
25     public: CMyClass(const CMyClass& iOther);
26     public: ~CMyClass();
27 
equal(const CMyClass & rhs) const28     bool equal(const CMyClass &rhs) const
29         { return fTag == rhs.fTag && fMagicValue == rhs.fMagicValue; }
30     private:
31         int fMagicValue;
32         int fTag;
33 
34     private: static int kStartedConstructionMagicValue;
35     private: static int kFinishedConstructionMagicValue;
36 };
37 
38 // Value for fMagicValue when the constructor has started running, but not yet finished
39 int CMyClass::kStartedConstructionMagicValue = 0;
40 // Value for fMagicValue when the constructor has finished running
41 int CMyClass::kFinishedConstructionMagicValue = 12345;
42 
CMyClass(int tag)43 CMyClass::CMyClass(int tag) :
44     fMagicValue(kStartedConstructionMagicValue), fTag(tag)
45 {
46     // Signal that the constructor has finished running
47     fMagicValue = kFinishedConstructionMagicValue;
48 }
49 
CMyClass(const CMyClass & iOther)50 CMyClass::CMyClass(const CMyClass& iOther) :
51     fMagicValue(kStartedConstructionMagicValue), fTag(iOther.fTag)
52 {
53     // If requested, throw an exception _before_ setting fMagicValue to kFinishedConstructionMagicValue
54     if (gCopyConstructorShouldThow) {
55         throw std::exception();
56     }
57     // Signal that the constructor has finished running
58     fMagicValue = kFinishedConstructionMagicValue;
59 }
60 
~CMyClass()61 CMyClass::~CMyClass() {
62     // Only instances for which the constructor has finished running should be destructed
63     assert(fMagicValue == kFinishedConstructionMagicValue);
64 }
65 
operator ==(const CMyClass & lhs,const CMyClass & rhs)66 bool operator==(const CMyClass &lhs, const CMyClass &rhs) { return lhs.equal(rhs); }
67 
main()68 int main()
69 {
70     CMyClass instance(42);
71     std::vector<CMyClass> vec;
72 
73     vec.push_back(instance);
74     std::vector<CMyClass> vec2(vec);
75     assert(is_contiguous_container_asan_correct(vec));
76     assert(is_contiguous_container_asan_correct(vec2));
77 
78     gCopyConstructorShouldThow = true;
79     try {
80         vec.push_back(instance);
81     }
82     catch (...) {
83         assert(vec==vec2);
84         assert(is_contiguous_container_asan_correct(vec));
85     }
86 }
87