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
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
|
/**
* @file Bug_3911_Regression_Test.cpp
*
* Reproduces the problem reported in bug 3911
* http://bugzilla.dre.vanderbilt.edu/show_bug.cgi?id=3911
*/
#include "test_config.h"
#include <ace/Malloc_T.h>
#include <ace/Memory_Pool.h>
#include <ace/Process_Mutex.h>
#include <ace/Null_Mutex.h>
#include <ace/PI_Malloc.h>
#define MMAP_FILENAME ACE_TEXT ("shared_memory")
static void
init_test ()
{
// Cleanup the MMAP file so we won't trip over the leftover mmap
// file from the previous crash.
ACE_MMAP_Memory_Pool_Options options (ACE_DEFAULT_BASE_ADDR);
//FUZZ: disable check_for_lack_ACE_OS
ACE_MMAP_Memory_Pool mmap (MMAP_FILENAME, &options);
//FUZZ: enable check_for_lack_ACE_OS
size_t rbyte = 0;
int ft = 0;
mmap.init_acquire (1024, rbyte, ft);
mmap.release ();
}
class ShmemMan
{
private:
// Used to implement the singleton pattern:
static ShmemMan* c_instance;
// The memory pool managing all the shared memory:
ACE_Malloc_T<ACE_MMAP_MEMORY_POOL,
ACE_Null_Mutex,
ACE_PI_Control_Block>* c_memory_pool;
protected:
ShmemMan(bool no_crash)
{
if (no_crash)
{
ACE_MMAP_Memory_Pool_Options options (ACE_DEFAULT_BASE_ADDR);
c_memory_pool = new ACE_Malloc_T<ACE_MMAP_MEMORY_POOL,
ACE_Null_Mutex,
ACE_PI_Control_Block>(MMAP_FILENAME
, ACE_TEXT(""), &options
);
}
else
{
c_memory_pool = new ACE_Malloc_T<ACE_MMAP_MEMORY_POOL,
ACE_Null_Mutex,
ACE_PI_Control_Block>(MMAP_FILENAME);
}
};
public:
~ShmemMan()
{
c_memory_pool->release();
delete c_memory_pool;
};
static ShmemMan* getInstance(bool no_crash)
{
if (c_instance == 0)
{
c_instance = new ShmemMan(no_crash);
}
return c_instance;
};
int clean()
{
c_memory_pool->release();
c_memory_pool->remove();
return 0;
};
void* getMemoryBlock(const char* block_name, unsigned int block_size)
{
void* shared = 0;
ACE_DEBUG((LM_INFO, ACE_TEXT("errno = %d. Looking for a Shared Memory block named %C\n"),
ACE_OS::last_error(),
block_name));
if (c_memory_pool->find(block_name, shared) == 0)
{
// An existing block was found, so take that:
ACE_DEBUG((LM_INFO, ACE_TEXT("Shared Memory block %C was found."),
block_name));
}
else
{
// Allocate the memory and bind it to a name:
ACE_DEBUG((LM_INFO, ACE_TEXT("Shared Memory block %C was not found. errno = %d. Trying to allocate new block\n"),
block_name,
ACE_OS::last_error()));
shared = c_memory_pool->malloc(block_size);
// reinterpret_cast due to some compilers warning against ordering pointers with integers
if (ptrdiff_t (shared) < 0)
{
ACE_DEBUG((LM_INFO, ACE_TEXT("New Shared Memory block could not be allocated. errno = %d.\n"),
ACE_OS::last_error()));
return (void*)(-1);
}
ACE_DEBUG((LM_INFO, ACE_TEXT("New Shared Memory block was allocated, trying to bind it to the name %C\n"),
block_name));
if (c_memory_pool->bind(block_name, shared) < 0)
{
ACE_DEBUG((LM_INFO, ACE_TEXT("New Shared Memory block could not be bound to the name %C. errno = %d.\n"),
block_name,
ACE_OS::last_error()));
return (void*)(-1);
}
}
return shared;
};
};
ShmemMan* ShmemMan::c_instance = 0;
const char* block_name = "block_1";
// Main function.
int
run_main (int argc, ACE_TCHAR * argv[])
{
ACE_START_TEST (ACE_TEXT ("Bug_3911_Regression_Test"));
init_test ();
bool no_crash = (argc>1 && argv[1][0]=='1');
ShmemMan* smm = ShmemMan::getInstance (no_crash);
void* buf = smm->getMemoryBlock (block_name, 10 * 4096);
ACE_DEBUG((LM_INFO, ACE_TEXT("allocated shmem block at %@\n"), buf));
smm->clean ();
ACE_END_TEST;
return 0;
}
|
