110 lines
3.0 KiB
C++
110 lines
3.0 KiB
C++
|
/*
|
||
|
|
* 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");
|
||
|
|
}
|
||
|
|
|