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
|
/*
* Copyright (C) 2014 The Android Open Source Project
*
* 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.
*/
#include <gtest/gtest.h>
#include <dlfcn.h>
#include <libgen.h>
#include <limits.h>
#include <stdio.h>
#include <stdint.h>
#include <string>
TEST(atexit, sofile) {
std::string atexit_call_sequence;
bool valid_this_in_static_dtor = false;
bool attr_dtor_called = false;
void* handle = dlopen("libtest_atexit.so", RTLD_NOW);
ASSERT_TRUE(handle != nullptr);
typedef int (*int_fn)(void);
int_fn get_cxx_ctor_called, get_attr_ctor_called;
get_cxx_ctor_called = reinterpret_cast<int_fn>(dlsym(handle, "get_cxx_ctor_called"));
get_attr_ctor_called = reinterpret_cast<int_fn>(dlsym(handle, "get_attr_ctor_called"));
ASSERT_TRUE(get_cxx_ctor_called != nullptr);
ASSERT_TRUE(get_attr_ctor_called != nullptr);
ASSERT_EQ(1, get_cxx_ctor_called());
ASSERT_EQ(1, get_attr_ctor_called());
void* sym = dlsym(handle, "register_atexit");
ASSERT_TRUE(sym != nullptr);
reinterpret_cast<void (*)(std::string*, bool*, bool*)>(sym)(&atexit_call_sequence, &valid_this_in_static_dtor, &attr_dtor_called);
ASSERT_EQ(0, dlclose(handle));
// this test verifies atexit call from atexit handler. as well as the order of calls
ASSERT_EQ("Humpty Dumpty sat on a wall", atexit_call_sequence);
ASSERT_TRUE(valid_this_in_static_dtor);
ASSERT_TRUE(attr_dtor_called);
}
class TestMainStaticDtorClass {
public:
TestMainStaticDtorClass() {
expected_this = this;
}
~TestMainStaticDtorClass() {
if (this != expected_this) {
fprintf(stderr, "\nerror: static d-tor called with incorrect this pointer: %p, expected: %p\n", this, expected_this);
} else {
fprintf(stderr, "6");
}
}
private:
static const TestMainStaticDtorClass* expected_this;
};
const TestMainStaticDtorClass* TestMainStaticDtorClass::expected_this = nullptr;
static void atexit_func5() {
fprintf(stderr, "5");
}
static void atexit_func4() {
fprintf(stderr, "4");
}
static void atexit_func3() {
fprintf(stderr, "3");
atexit(atexit_func4);
}
static void atexit_func2() {
fprintf(stderr, "2");
}
static void atexit_func1() {
fprintf(stderr, "1");
}
static void atexit_main() {
// This should result in "123456" output to stderr
static TestMainStaticDtorClass static_obj;
atexit(atexit_func5);
atexit(atexit_func3);
atexit(atexit_func2);
atexit(atexit_func1);
exit(0);
}
TEST(atexit, exit) {
ASSERT_EXIT(atexit_main(), testing::ExitedWithCode(0), "123456");
}
|
