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 // <functional>
11 
12 // class function<R(ArgTypes...)>
13 
14 // function(const function& f);
15 
16 #include <functional>
17 #include <cstdlib>
18 #include <cassert>
19 
20 #include "count_new.hpp"
21 
22 class A
23 {
24     int data_[10];
25 public:
26     static int count;
27 
A()28     A()
29     {
30         ++count;
31         for (int i = 0; i < 10; ++i)
32             data_[i] = i;
33     }
34 
A(const A &)35     A(const A&) {++count;}
36 
~A()37     ~A() {--count;}
38 
operator ()(int i) const39     int operator()(int i) const
40     {
41         for (int j = 0; j < 10; ++j)
42             i += data_[j];
43         return i;
44     }
45 };
46 
47 int A::count = 0;
48 
g(int)49 int g(int) {return 0;}
50 
main()51 int main()
52 {
53     assert(globalMemCounter.checkOutstandingNewEq(0));
54     {
55     std::function<int(int)> f = A();
56     assert(A::count == 1);
57     assert(globalMemCounter.checkOutstandingNewEq(1));
58     assert(f.target<A>());
59     assert(f.target<int(*)(int)>() == 0);
60     std::function<int(int)> f2 = f;
61     assert(A::count == 2);
62     assert(globalMemCounter.checkOutstandingNewEq(2));
63     assert(f2.target<A>());
64     assert(f2.target<int(*)(int)>() == 0);
65     }
66     assert(A::count == 0);
67     assert(globalMemCounter.checkOutstandingNewEq(0));
68     {
69     std::function<int(int)> f = g;
70     assert(globalMemCounter.checkOutstandingNewEq(0));
71     assert(f.target<int(*)(int)>());
72     assert(f.target<A>() == 0);
73     std::function<int(int)> f2 = f;
74     assert(globalMemCounter.checkOutstandingNewEq(0));
75     assert(f2.target<int(*)(int)>());
76     assert(f2.target<A>() == 0);
77     }
78     assert(globalMemCounter.checkOutstandingNewEq(0));
79     {
80     std::function<int(int)> f;
81     assert(globalMemCounter.checkOutstandingNewEq(0));
82     assert(f.target<int(*)(int)>() == 0);
83     assert(f.target<A>() == 0);
84     std::function<int(int)> f2 = f;
85     assert(globalMemCounter.checkOutstandingNewEq(0));
86     assert(f2.target<int(*)(int)>() == 0);
87     assert(f2.target<A>() == 0);
88     }
89     {
90     std::function<int(int)> f;
91     assert(globalMemCounter.checkOutstandingNewEq(0));
92     assert(f.target<int(*)(int)>() == 0);
93     assert(f.target<A>() == 0);
94     assert(!f);
95     std::function<long(int)> g = f;
96     assert(globalMemCounter.checkOutstandingNewEq(0));
97     assert(g.target<long(*)(int)>() == 0);
98     assert(g.target<A>() == 0);
99     assert(!g);
100     }
101 #ifndef _LIBCPP_HAS_NO_RVALUE_REFERENCES
102     assert(globalMemCounter.checkOutstandingNewEq(0));
103     {
104     std::function<int(int)> f = A();
105     assert(A::count == 1);
106     assert(globalMemCounter.checkOutstandingNewEq(1));
107     assert(f.target<A>());
108     assert(f.target<int(*)(int)>() == 0);
109     std::function<int(int)> f2 = std::move(f);
110     assert(A::count == 1);
111     assert(globalMemCounter.checkOutstandingNewEq(1));
112     assert(f2.target<A>());
113     assert(f2.target<int(*)(int)>() == 0);
114     assert(f.target<A>() == 0);
115     assert(f.target<int(*)(int)>() == 0);
116     }
117 #endif  // _LIBCPP_HAS_NO_RVALUE_REFERENCES
118 }
119