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 #include <gtest/gtest.h>
  18
  19 #include <dlfcn.h>
  20 #include <libgen.h>
  21 #include <limits.h>
  22 #include <stdio.h>
  23 #include <stdint.h>
  24
  25 #include <string>
  26
  27 TEST(atexit, dlclose) {
  28   std::string atexit_call_sequence;
  29   bool valid_this_in_static_dtor = false;
  30   void* handle = dlopen("libtest_atexit.so", RTLD_NOW);
  31   ASSERT_TRUE(handle != NULL);
  32
  33   void* sym = dlsym(handle, "register_atexit");
  34   ASSERT_TRUE(sym != NULL);
  35   reinterpret_cast<void (*)(std::string*, bool*)>(sym)(&atexit_call_sequence, &valid_this_in_static_dtor);
  36
  37   ASSERT_EQ(0, dlclose(handle));
  38   // this test verifies atexit call from atexit handler. as well as the order of calls
  39   ASSERT_EQ("Humpty Dumpty sat on a wall", atexit_call_sequence);
  40   ASSERT_TRUE(valid_this_in_static_dtor);
  41 }
  42
  43 class TestMainStaticDtorClass {
  44  public:
  45   TestMainStaticDtorClass() {
  46     expected_this = this;
  47   }
  48
  49   ~TestMainStaticDtorClass() {
  50     if (this != expected_this) {
  51       fprintf(stderr, "\nerror: static d-tor called with incorrect this pointer: %p, expected: %p\n", this, expected_this);
  52     } else {
  53       fprintf(stderr, "6");
  54     }
  55   }
  56  private:
  57   static const TestMainStaticDtorClass* expected_this;
  58 };
  59
  60 const TestMainStaticDtorClass* TestMainStaticDtorClass::expected_this = NULL;
  61
  62 static void atexit_func5() {
  63   fprintf(stderr, "5");
  64 }
  65
  66 static void atexit_func4() {
  67   fprintf(stderr, "4");
  68 }
  69
  70 static void atexit_func3() {
  71   fprintf(stderr, "3");
  72   atexit(atexit_func4);
  73 }
  74
  75 static void atexit_func2() {
  76   fprintf(stderr, "2");
  77 }
  78
  79 static void atexit_func1() {
  80   fprintf(stderr, "1");
  81 }
  82
  83 static void atexit_main() {
  84   // This should result in "123456" output to stderr
  85   static TestMainStaticDtorClass static_obj;
  86   atexit(atexit_func5);
  87   atexit(atexit_func3);
  88   atexit(atexit_func2);
  89   atexit(atexit_func1);
  90   exit(0);
  91 }
  92
  93 TEST(atexit, exit) {
  94   ASSERT_EXIT(atexit_main(), testing::ExitedWithCode(0), "123456");
  95 }
  96