1 /*
2 * Copyright (C) 2014 The Android Open Source Project
3 *
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 * http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 */
16
17 // To work around b/25643775, we disable clang optimization so that
18 // VTT for std::__1::basic_stringstream<char, std::__1::char_traits<char>,
19 // std::__1::allocator<char> >
20 // will be correctly kept for other module's references.
21 #if defined(__clang__) && (defined(__arm__) || defined(__aarch64__))
22 #pragma clang optimize off
23 #endif
24 #include <gtest/gtest.h>
25 #if defined(__clang__) && (defined(__arm__) || defined(__aarch64__))
26 #pragma clang optimize on
27 #endif
28
29 #include <dlfcn.h>
30 #include <libgen.h>
31 #include <limits.h>
32 #include <stdio.h>
33 #include <stdint.h>
34
35 #include <string>
36
TEST(atexit,sofile)37 TEST(atexit, sofile) {
38 std::string atexit_call_sequence;
39 bool valid_this_in_static_dtor = false;
40 bool attr_dtor_called = false;
41
42 void* handle = dlopen("libtest_atexit.so", RTLD_NOW);
43 ASSERT_TRUE(handle != nullptr);
44
45 typedef int (*int_fn)(void);
46 int_fn get_cxx_ctor_called, get_attr_ctor_called;
47 get_cxx_ctor_called = reinterpret_cast<int_fn>(dlsym(handle, "get_cxx_ctor_called"));
48 get_attr_ctor_called = reinterpret_cast<int_fn>(dlsym(handle, "get_attr_ctor_called"));
49 ASSERT_TRUE(get_cxx_ctor_called != nullptr);
50 ASSERT_TRUE(get_attr_ctor_called != nullptr);
51
52 ASSERT_EQ(1, get_cxx_ctor_called());
53 ASSERT_EQ(1, get_attr_ctor_called());
54
55 void* sym = dlsym(handle, "register_atexit");
56 ASSERT_TRUE(sym != nullptr);
57 reinterpret_cast<void (*)(std::string*, bool*, bool*)>(sym)(&atexit_call_sequence, &valid_this_in_static_dtor, &attr_dtor_called);
58
59 ASSERT_EQ(0, dlclose(handle));
60 // this test verifies atexit call from atexit handler. as well as the order of calls
61 ASSERT_EQ("Humpty Dumpty sat on a wall", atexit_call_sequence);
62 ASSERT_TRUE(valid_this_in_static_dtor);
63 ASSERT_TRUE(attr_dtor_called);
64 }
65
66 class TestMainStaticDtorClass {
67 public:
TestMainStaticDtorClass()68 TestMainStaticDtorClass() {
69 expected_this = this;
70 }
71
~TestMainStaticDtorClass()72 ~TestMainStaticDtorClass() {
73 if (this != expected_this) {
74 fprintf(stderr, "\nerror: static d-tor called with incorrect this pointer: %p, expected: %p\n", this, expected_this);
75 } else {
76 fprintf(stderr, "6");
77 }
78 }
79 private:
80 static const TestMainStaticDtorClass* expected_this;
81 };
82
83 const TestMainStaticDtorClass* TestMainStaticDtorClass::expected_this = nullptr;
84
atexit_func5()85 static void atexit_func5() {
86 fprintf(stderr, "5");
87 }
88
atexit_func4()89 static void atexit_func4() {
90 fprintf(stderr, "4");
91 }
92
atexit_func3()93 static void atexit_func3() {
94 fprintf(stderr, "3");
95 atexit(atexit_func4);
96 }
97
atexit_func2()98 static void atexit_func2() {
99 fprintf(stderr, "2");
100 }
101
atexit_func1()102 static void atexit_func1() {
103 fprintf(stderr, "1");
104 }
105
atexit_main()106 static void atexit_main() {
107 // This should result in "123456" output to stderr
108 static TestMainStaticDtorClass static_obj;
109 atexit(atexit_func5);
110 atexit(atexit_func3);
111 atexit(atexit_func2);
112 atexit(atexit_func1);
113 exit(0);
114 }
115
TEST(atexit,exit)116 TEST(atexit, exit) {
117 ASSERT_EXIT(atexit_main(), testing::ExitedWithCode(0), "123456");
118 }
119
120