1 #include <boost/config.hpp>
2 
3 #if defined(BOOST_MSVC)
4 #pragma warning(disable: 4786)  // identifier truncated in debug info
5 #pragma warning(disable: 4710)  // function not inlined
6 #pragma warning(disable: 4711)  // function selected for automatic inline expansion
7 #pragma warning(disable: 4514)  // unreferenced inline removed
8 #endif
9 
10 //  weak_ptr_mt_test.cpp
11 //
12 //  Copyright (c) 2002 Peter Dimov and Multi Media Ltd.
13 //  Copyright 2005, 2008 Peter Dimov
14 //
15 //  Distributed under the Boost Software License, Version 1.0.
16 //  See accompanying file LICENSE_1_0.txt or copy at
17 //  http://www.boost.org/LICENSE_1_0.txt
18 
19 #include <boost/shared_ptr.hpp>
20 #include <boost/weak_ptr.hpp>
21 #include <boost/bind.hpp>
22 
23 #include <vector>
24 
25 #include <cstdio>
26 #include <ctime>
27 #include <cstdlib>
28 
29 #include <boost/detail/lightweight_thread.hpp>
30 
31 //
32 
33 int const n = 16384;
34 int const k = 512; // vector size
35 int const m = 16; // threads
36 
test(std::vector<boost::shared_ptr<int>> & v)37 void test( std::vector< boost::shared_ptr<int> > & v )
38 {
39     using namespace std; // printf, rand
40 
41     std::vector< boost::weak_ptr<int> > w( v.begin(), v.end() );
42 
43     int s = 0, f = 0, r = 0;
44 
45     for( int i = 0; i < n; ++i )
46     {
47         // randomly kill a pointer
48 
49         v[ rand() % k ].reset();
50         ++s;
51 
52         for( int j = 0; j < k; ++j )
53         {
54             if( boost::shared_ptr<int> px = w[ j ].lock() )
55             {
56                 ++s;
57 
58                 if( rand() & 4 )
59                 {
60                     continue;
61                 }
62 
63                 // rebind anyway with prob. 50% for add_ref_lock() against weak_release() contention
64                 ++f;
65             }
66             else
67             {
68                 ++r;
69             }
70 
71             w[ j ] = v[ rand() % k ];
72         }
73     }
74 
75     printf( "\n%d locks, %d forced rebinds, %d normal rebinds.", s, f, r );
76 }
77 
78 #if defined( BOOST_HAS_PTHREADS )
79 
80 char const * thmodel = "POSIX";
81 
82 #else
83 
84 char const * thmodel = "Windows";
85 
86 #endif
87 
main()88 int main()
89 {
90     using namespace std; // printf, clock_t, clock
91 
92     printf("Using %s threads: %d threads, %d * %d iterations: ", thmodel, m, n, k );
93 
94     std::vector< boost::shared_ptr<int> > v( k );
95 
96     for( int i = 0; i < k; ++i )
97     {
98         v[ i ].reset( new int( 0 ) );
99     }
100 
101     clock_t t = clock();
102 
103     boost::detail::lw_thread_t a[ m ];
104 
105     for( int i = 0; i < m; ++i )
106     {
107         boost::detail::lw_thread_create( a[ i ], boost::bind( test, v ) );
108     }
109 
110     v.resize( 0 ); // kill original copies
111 
112     for( int j = 0; j < m; ++j )
113     {
114         boost::detail::lw_thread_join( a[j] );
115     }
116 
117     t = clock() - t;
118 
119     printf("\n\n%.3f seconds.\n", static_cast<double>(t) / CLOCKS_PER_SEC);
120 
121     return 0;
122 }
123