1 #ifndef TEST_ALLOCATOR_H 2 #define TEST_ALLOCATOR_H 3 4 #include <cstddef> 5 #include <type_traits> 6 #include <cstdlib> 7 #include <new> 8 #include <climits> 9 10 class test_alloc_base 11 { 12 protected: 13 static int count; 14 public: 15 static int throw_after; 16 }; 17 18 int test_alloc_base::count = 0; 19 int test_alloc_base::throw_after = INT_MAX; 20 21 template <class T> 22 class test_allocator 23 : public test_alloc_base 24 { 25 int data_; 26 27 template <class U> friend class test_allocator; 28 public: 29 30 typedef unsigned size_type; 31 typedef int difference_type; 32 typedef T value_type; 33 typedef value_type* pointer; 34 typedef const value_type* const_pointer; 35 typedef typename std::add_lvalue_reference<value_type>::type reference; 36 typedef typename std::add_lvalue_reference<const value_type>::type const_reference; 37 38 template <class U> struct rebind {typedef test_allocator<U> other;}; 39 throw()40 test_allocator() throw() : data_(-1) {} test_allocator(int i)41 explicit test_allocator(int i) throw() : data_(i) {} throw()42 test_allocator(const test_allocator& a) throw() 43 : data_(a.data_) {} test_allocator(const test_allocator<U> & a)44 template <class U> test_allocator(const test_allocator<U>& a) throw() 45 : data_(a.data_) {} throw()46 ~test_allocator() throw() {data_ = 0;} address(reference x)47 pointer address(reference x) const {return &x;} address(const_reference x)48 const_pointer address(const_reference x) const {return &x;} 49 pointer allocate(size_type n, const void* = 0) 50 { 51 if (count >= throw_after) 52 throw std::bad_alloc(); 53 ++count; 54 return (pointer)std::malloc(n * sizeof(T)); 55 } deallocate(pointer p,size_type n)56 void deallocate(pointer p, size_type n) 57 {std::free(p);} max_size()58 size_type max_size() const throw() 59 {return UINT_MAX / sizeof(T);} construct(pointer p,const T & val)60 void construct(pointer p, const T& val) 61 {::new(p) T(val);} 62 #ifndef _LIBCPP_HAS_NO_RVALUE_REFERENCES construct(pointer p,T && val)63 void construct(pointer p, T&& val) 64 {::new(p) T(std::move(val));} 65 #endif // _LIBCPP_HAS_NO_RVALUE_REFERENCES destroy(pointer p)66 void destroy(pointer p) {p->~T();} 67 68 friend bool operator==(const test_allocator& x, const test_allocator& y) 69 {return x.data_ == y.data_;} 70 friend bool operator!=(const test_allocator& x, const test_allocator& y) 71 {return !(x == y);} 72 }; 73 74 template <class T> 75 class other_allocator 76 { 77 int data_; 78 79 template <class U> friend class other_allocator; 80 81 public: 82 typedef T value_type; 83 other_allocator()84 other_allocator() : data_(-1) {} other_allocator(int i)85 explicit other_allocator(int i) : data_(i) {} other_allocator(const other_allocator<U> & a)86 template <class U> other_allocator(const other_allocator<U>& a) 87 : data_(a.data_) {} allocate(std::size_t n)88 T* allocate(std::size_t n) 89 {return (T*)std::malloc(n * sizeof(T));} deallocate(T * p,std::size_t n)90 void deallocate(T* p, std::size_t n) 91 {std::free(p);} 92 select_on_container_copy_construction()93 other_allocator select_on_container_copy_construction() const 94 {return other_allocator(-2);} 95 96 friend bool operator==(const other_allocator& x, const other_allocator& y) 97 {return x.data_ == y.data_;} 98 friend bool operator!=(const other_allocator& x, const other_allocator& y) 99 {return !(x == y);} 100 101 typedef std::true_type propagate_on_container_copy_assignment; 102 typedef std::true_type propagate_on_container_move_assignment; 103 typedef std::true_type propagate_on_container_swap; 104 105 #ifdef _LIBCPP_HAS_NO_ADVANCED_SFINAE max_size()106 std::size_t max_size() const 107 {return UINT_MAX / sizeof(T);} 108 #endif // _LIBCPP_HAS_NO_ADVANCED_SFINAE 109 110 }; 111 112 #endif // TEST_ALLOCATOR_H 113