Apply patch for [CVE-2012-2677][boost] ordered_malloc() overflow
[external/boost.git] / libs / smart_ptr / test / sp_atomic_test.cpp
1 #include <boost/config.hpp>
2
3 //  sp_atomic_test.cpp
4 //
5 //  Copyright (c) 2008 Peter Dimov
6 //
7 //  Distributed under the Boost Software License, Version 1.0.
8 //  See accompanying file LICENSE_1_0.txt or copy at
9 //  http://www.boost.org/LICENSE_1_0.txt
10
11
12 #include <boost/detail/lightweight_test.hpp>
13 #include <boost/shared_ptr.hpp>
14
15 //
16
17 struct X
18 {
19 };
20
21 #define BOOST_TEST_SP_EQ( p, q ) BOOST_TEST( p == q && !( p < q ) && !( q < p ) )
22
23 int main()
24 {
25     boost::shared_ptr<X> px( new X );
26
27     {
28         boost::shared_ptr<X> p2 = boost::atomic_load( &px );
29         BOOST_TEST_SP_EQ( p2, px );
30
31         boost::shared_ptr<X> px2( new X );
32         boost::atomic_store( &px, px2 );
33         BOOST_TEST_SP_EQ( px, px2 );
34
35         p2 = boost::atomic_load( &px );
36         BOOST_TEST_SP_EQ( p2, px );
37         BOOST_TEST_SP_EQ( p2, px2 );
38
39         boost::shared_ptr<X> px3( new X );
40         boost::shared_ptr<X> p3 = boost::atomic_exchange( &px, px3 );
41         BOOST_TEST_SP_EQ( p3, px2 );
42         BOOST_TEST_SP_EQ( px, px3 );
43
44         boost::shared_ptr<X> px4( new X );
45         boost::shared_ptr<X> cmp;
46
47         bool r = boost::atomic_compare_exchange( &px, &cmp, px4 );
48         BOOST_TEST( !r );
49         BOOST_TEST_SP_EQ( px, px3 );
50         BOOST_TEST_SP_EQ( cmp, px3 );
51
52         r = boost::atomic_compare_exchange( &px, &cmp, px4 );
53         BOOST_TEST( r );
54         BOOST_TEST_SP_EQ( px, px4 );
55     }
56
57     //
58
59     px.reset();
60
61     {
62         boost::shared_ptr<X> p2 = boost::atomic_load_explicit( &px, boost::memory_order_acquire );
63         BOOST_TEST_SP_EQ( p2, px );
64
65         boost::shared_ptr<X> px2( new X );
66         boost::atomic_store_explicit( &px, px2, boost::memory_order_release );
67         BOOST_TEST_SP_EQ( px, px2 );
68
69         boost::shared_ptr<X> p3 = boost::atomic_exchange_explicit( &px, boost::shared_ptr<X>(), boost::memory_order_acq_rel );
70         BOOST_TEST_SP_EQ( p3, px2 );
71         BOOST_TEST_SP_EQ( px, p2 );
72
73         boost::shared_ptr<X> px4( new X );
74         boost::shared_ptr<X> cmp( px2 );
75
76         bool r = boost::atomic_compare_exchange_explicit( &px, &cmp, px4, boost::memory_order_acquire, boost::memory_order_relaxed );
77         BOOST_TEST( !r );
78         BOOST_TEST_SP_EQ( px, p2 );
79         BOOST_TEST_SP_EQ( cmp, p2 );
80
81         r = boost::atomic_compare_exchange_explicit( &px, &cmp, px4, boost::memory_order_release, boost::memory_order_acquire );
82         BOOST_TEST( r );
83         BOOST_TEST_SP_EQ( px, px4 );
84     }
85
86     return boost::report_errors();
87 }