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
158
159
160
161
162
163
164
165
166
167
168
169
|
/*
Copyright (c) 2005-2024 Intel Corporation
Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at
http://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
*/
/* Regression test against a bug in TBB allocator manifested when
dynamic library calls atexit() or registers dtors of static objects.
If the allocator is not initialized yet, we can get deadlock,
because allocator library has static object dtors as well, they
registered during allocator initialization, and atexit() is protected
by non-recursive mutex in some versions of GLIBC.
*/
#define __TBB_NO_IMPLICIT_LINKAGE 1
#include "common/allocator_overload.h"
#include "common/utils_assert.h"
#include <stdlib.h>
// __TBB_malloc_safer_msize() returns 0 for unknown objects,
// thus we can detect ownership
#if _USRDLL
#if _WIN32||_WIN64
extern __declspec(dllexport)
#else
__TBB_EXPORT
#endif
bool dll_isMallocOverloaded()
#else
bool exe_isMallocOverloaded()
#endif
{
const size_t reqSz = 8;
void *o = malloc(reqSz);
bool ret = __TBB_malloc_safer_msize(o, nullptr) >= reqSz;
free(o);
return ret;
}
#if _USRDLL
#include "common/utils_report.h"
#if MALLOC_UNIXLIKE_OVERLOAD_ENABLED || MALLOC_ZONE_OVERLOAD_ENABLED
#include <dlfcn.h>
#if __APPLE__
#include <malloc/malloc.h>
#define malloc_usable_size(p) malloc_size(p)
#else
#include <malloc.h>
#endif
#include <signal.h>
#if __unix__ && !__ANDROID__
extern "C" {
void __libc_free(void *ptr);
void *__libc_realloc(void *ptr, size_t size);
// check that such kind of free/realloc overload works correctly
void free(void *ptr)
{
__libc_free(ptr);
}
void *realloc(void *ptr, size_t size)
{
return __libc_realloc(ptr, size);
}
} // extern "C"
#endif // __unix__ && !__ANDROID__
#endif // MALLOC_UNIXLIKE_OVERLOAD_ENABLED || MALLOC_ZONE_OVERLOAD_ENABLED
// Even when the test is skipped, dll source must not be empty to generate .lib to link with.
#if !defined(_PGO_INSTRUMENT) && !__TBB_USE_ADDRESS_SANITIZER
void dummyFunction() {}
// TODO: enable the check under Android
#if (MALLOC_UNIXLIKE_OVERLOAD_ENABLED || MALLOC_ZONE_OVERLOAD_ENABLED) && !__ANDROID__
typedef void *(malloc_type)(size_t);
static void SigSegv(int)
{
REPORT("Known issue: SIGSEGV during work with memory allocated by replaced allocator.\n"
"skip\n");
exit(0);
}
// TODO: Using of SIGSEGV can be eliminated via parsing /proc/self/maps
// and series of system malloc calls.
void TestReplacedAllocFunc()
{
struct sigaction sa, sa_default;
malloc_type *orig_malloc = (malloc_type*)dlsym(RTLD_NEXT, "malloc");
void *p = (*orig_malloc)(16);
// protect potentially unsafe actions
sigemptyset(&sa.sa_mask);
sa.sa_flags = 0;
sa.sa_handler = SigSegv;
if (sigaction(SIGSEGV, &sa, &sa_default))
ASSERT(0, "sigaction failed");
ASSERT(malloc_usable_size(p) >= 16, nullptr);
free(p);
// no more unsafe actions, restore SIGSEGV
if (sigaction(SIGSEGV, &sa_default, nullptr))
ASSERT(0, "sigaction failed");
}
#else
void TestReplacedAllocFunc() { }
#endif
class Foo {
public:
Foo() {
// add a lot of exit handlers to cause memory allocation
for (int i=0; i<1024; i++)
atexit(dummyFunction);
TestReplacedAllocFunc();
}
};
static Foo f;
#endif // !defined(_PGO_INSTRUMENT) && !__TBB_USE_ADDRESS_SANITIZER
int main() {}
#else // _USRDLL
#include "common/test.h"
#if _WIN32||_WIN64
#include "tbb/tbbmalloc_proxy.h"
extern __declspec(dllimport)
#endif
bool dll_isMallocOverloaded();
#ifdef _PGO_INSTRUMENT
//! \brief \ref error_guessing
TEST_CASE("Known issue: test_malloc_atexit hangs if compiled with -prof-genx\n" * doctest::skip(true)) {}
#elif __TBB_USE_ADDRESS_SANITIZER
//! \brief \ref error_guessing
TEST_CASE("Known issue: test_malloc_atexit is not applicable under ASAN\n" * doctest::skip(true)) {}
#else
// Check common/allocator_overload.h for skip cases
#if !HARNESS_SKIP_TEST
//! \brief \ref error_guessing
TEST_CASE("test malloc atexit") {
REQUIRE_MESSAGE( dll_isMallocOverloaded(), "malloc was not replaced" );
REQUIRE_MESSAGE( exe_isMallocOverloaded(), "malloc was not replaced" );
}
#endif // HARNESS_SKIP_TEST
#endif // _PGO_INSTRUMENT
#endif // _USRDLL
|
