tests/atexit_test.cpp

   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
  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:
  68   TestMainStaticDtorClass() {
  69     expected_this = this;
  70   }
  71
  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
  85 static void atexit_func5() {
  86   fprintf(stderr, "5");
  87 }
  88
  89 static void atexit_func4() {
  90   fprintf(stderr, "4");
  91 }
  92
  93 static void atexit_func3() {
  94   fprintf(stderr, "3");
  95   atexit(atexit_func4);
  96 }
  97
  98 static void atexit_func2() {
  99   fprintf(stderr, "2");
 100 }
 101
 102 static void atexit_func1() {
 103   fprintf(stderr, "1");
 104 }
 105
 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
 116 TEST(atexit, exit) {
 117   ASSERT_EXIT(atexit_main(), testing::ExitedWithCode(0), "123456");
 118 }
 119