1 //////////////////////////////////////////////////////////////////////////////
2 //
3 // (C) Copyright Ion Gaztanaga 2015-2015. Distributed under the Boost
4 // Software License, Version 1.0. (See accompanying file
5 // LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
6 //
7 // See http://www.boost.org/libs/container for documentation.
8 //
9 //////////////////////////////////////////////////////////////////////////////
10 #include <boost/container/pmr/global_resource.hpp>
11 #include <boost/container/pmr/memory_resource.hpp>
12 #include <boost/core/lightweight_test.hpp>
13 #include <boost/core/no_exceptions_support.hpp>
14 
15 #include "derived_from_memory_resource.hpp"
16 
17 #include <cstdlib>
18 #include <new>
19 
20 using namespace boost::container;
21 using namespace boost::container::pmr;
22 
23 #ifdef BOOST_MSVC
24 #pragma warning (push)
25 #pragma warning (disable : 4290)
26 #endif
27 
28 #if __cplusplus >= 201103L
29 #define BOOST_CONTAINER_NEW_EXCEPTION_SPECIFIER
30 #define BOOST_CONTAINER_DELETE_EXCEPTION_SPECIFIER noexcept
31 #else
32 #define BOOST_CONTAINER_NEW_EXCEPTION_SPECIFIER    throw(std::bad_alloc)
33 #define BOOST_CONTAINER_DELETE_EXCEPTION_SPECIFIER throw()
34 #endif
35 
36 #if defined(BOOST_GCC) && (BOOST_GCC >= 50000)
37 #pragma GCC diagnostic ignored "-Wsized-deallocation"
38 #endif
39 
40 //ASAN does not support operator new overloading
41 #ifndef BOOST_CONTAINER_ASAN
42 
43 std::size_t allocation_count = 0;
44 
operator new[](std::size_t count)45 void* operator new[](std::size_t count) BOOST_CONTAINER_NEW_EXCEPTION_SPECIFIER
46 {
47    ++allocation_count;
48    return std::malloc(count);
49 }
50 
operator delete[](void * p)51 void operator delete[](void *p) BOOST_CONTAINER_DELETE_EXCEPTION_SPECIFIER
52 {
53    --allocation_count;
54    return std::free(p);
55 }
56 
57 #endif   //BOOST_CONTAINER_ASAN
58 
59 #ifdef BOOST_MSVC
60 #pragma warning (pop)
61 #endif
62 
63 #ifndef BOOST_CONTAINER_ASAN
64 
test_new_delete_resource()65 void test_new_delete_resource()
66 {
67    //Make sure new_delete_resource calls new[]/delete[]
68    std::size_t memcount = allocation_count;
69    memory_resource *mr = new_delete_resource();
70    //each time should return the same pointer
71    BOOST_TEST(mr == new_delete_resource());
72    #if !defined(BOOST_CONTAINER_DYNAMIC_LINKING)  //No new delete replacement possible new_delete is a DLL
73    BOOST_TEST(memcount == allocation_count);
74    #endif
75    void *addr = mr->allocate(16, 1);
76    #if !defined(BOOST_CONTAINER_DYNAMIC_LINKING)  //No new delete replacement possible new_delete is a DLL
77    BOOST_TEST((allocation_count - memcount) == 1);
78    #endif
79    mr->deallocate(addr, 16, 1);
80    BOOST_TEST(memcount == allocation_count);
81 }
82 
83 #endif   //BOOST_CONTAINER_ASAN
84 
test_null_memory_resource()85 void test_null_memory_resource()
86 {
87    //Make sure it throw or returns null
88    memory_resource *mr = null_memory_resource();
89    BOOST_TEST(mr != 0);
90 
91    #if !defined(BOOST_NO_EXCEPTIONS)
92    bool bad_allocexception_thrown = false;
93 
94    BOOST_TRY{
95       mr->allocate(1, 1);
96    }
97    BOOST_CATCH(std::bad_alloc&) {
98       bad_allocexception_thrown = true;
99    }
100    BOOST_CATCH(...) {
101    }
102    BOOST_CATCH_END
103 
104    BOOST_TEST(bad_allocexception_thrown == true);
105    #endif   //BOOST_NO_EXCEPTIONS
106 }
107 
test_default_resource()108 void test_default_resource()
109 {
110    //Default resource must be new/delete before set_default_resource
111    BOOST_TEST(get_default_resource() == new_delete_resource());
112    //Set default resource and obtain previous
113    derived_from_memory_resource d;
114    memory_resource *prev_default = set_default_resource(&d);
115    BOOST_TEST(get_default_resource() == &d);
116    //Set default resource with null, which should be new/delete
117    prev_default = set_default_resource(0);
118    BOOST_TEST(prev_default == &d);
119    BOOST_TEST(get_default_resource() == new_delete_resource());
120 }
121 
main()122 int main()
123 {
124    #ifndef BOOST_CONTAINER_ASAN
125    test_new_delete_resource();
126    #endif
127    test_null_memory_resource();
128    test_default_resource();
129    return ::boost::report_errors();
130 }
131