1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
|
//=============================================================================
/**
* @file Dynamic_Test.cpp
*
* This tests the ACE_Dynamic class
*
* @author Johnny Willemsen <jwillemsen@remedy.nl>
*/
//=============================================================================
#include "test_config.h"
#include "ace/OS_NS_string.h"
#include "ace/Dynamic.h"
#include "ace/OS_Memory.h"
class A
{
public:
A ();
void *operator new (size_t n);
void *operator new (size_t n, const std::nothrow_t&) noexcept;
void operator delete (void *p, const std::nothrow_t&) noexcept;
void * operator new (size_t n, void *p);
void operator delete (void *);
void operator delete (void *, void *);
/// Have we been dynamically created?
bool dynamic_;
};
void*
A::operator new (size_t n)
{
ACE_Dynamic *const dynamic_instance = ACE_Dynamic::instance ();
if (dynamic_instance == 0)
{
// If this ACE_TEST_ASSERT fails, it may be due to running of out TSS
// keys. Try using ACE_HAS_TSS_EMULATION, or increasing
// ACE_DEFAULT_THREAD_KEYS if already using TSS emulation.
ACE_TEST_ASSERT (dynamic_instance != 0);
ACE_throw_bad_alloc;
}
else
{
// Allocate the memory and store it (usually in thread-specific
// storage, depending on config flags).
dynamic_instance->set ();
return ::new char[n];
}
}
void*
A::operator new (size_t n, const std::nothrow_t&) noexcept
{
ACE_Dynamic *const dynamic_instance = ACE_Dynamic::instance ();
if (dynamic_instance == 0)
{
// If this ACE_TEST_ASSERT fails, it may be due to running of out TSS
// keys. Try using ACE_HAS_TSS_EMULATION, or increasing
// ACE_DEFAULT_THREAD_KEYS if already using TSS emulation.
ACE_TEST_ASSERT (dynamic_instance != 0);
return 0;
}
else
{
// Allocate the memory and store it (usually in thread-specific
// storage, depending on config flags).
dynamic_instance->set ();
return ::new(std::nothrow) char[n];
}
}
void
A::operator delete (void *p, const std::nothrow_t&) noexcept
{
::delete [] static_cast <char *> (p);
}
void
A::operator delete (void *obj)
{
::delete [] static_cast <char *> (obj);
}
A::A()
{
this->dynamic_ = ACE_Dynamic::instance ()->is_dynamic ();
if (this->dynamic_)
// Make sure to reset the flag.
ACE_Dynamic::instance ()->reset ();
}
int
run_main (int, ACE_TCHAR *[])
{
ACE_START_TEST (ACE_TEXT ("Dynamic_Test"));
A from_stack;
A* heap = 0;
ACE_NEW_RETURN (heap, A, 1);
if (from_stack.dynamic_)
{
ACE_ERROR_RETURN ((LM_ERROR,
ACE_TEXT ("dynamic_ is true for an object on the stack\n")),
1);
}
if (!heap->dynamic_)
{
ACE_ERROR_RETURN ((LM_ERROR,
ACE_TEXT ("dynamic_ is false for an object from the heap\n")),
1);
}
delete heap;
ACE_END_TEST;
return 0;
}
|
