xref: /aosp_15_r20/bionic/tests/libs/atexit_testlib.cpp (revision 8d67ca893c1523eb926b9080dbe4e2ffd2a27ba1)

/*
 * 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 <stdio.h>
#include <stdlib.h>

#include <string>

// use external control number from main test
static std::string* atexit_sequence = nullptr;
static bool* atexit_valid_this_in_static_dtor = nullptr;
static bool* atexit_attr_dtor_called = nullptr;

static int cxx_ctor_called = 0;
static int attr_ctor_called = 0;

static class AtExitStaticClass {
 public:
  AtExitStaticClass() {
    expected_this = this;
    cxx_ctor_called = 1;
  }
  ~AtExitStaticClass() {
    if (atexit_valid_this_in_static_dtor) {
      *atexit_valid_this_in_static_dtor = (expected_this == this);
    }
  }
 private:
  static const AtExitStaticClass* expected_this;

} static_obj;

const AtExitStaticClass* AtExitStaticClass::expected_this = nullptr;

// 4
static void atexit_handler_from_atexit_from_atexit2() {
  *atexit_sequence += " on";
}

// 3
static void atexit_handler_from_atexit_from_atexit1() {
  *atexit_sequence += " sat";
}

// 2
static void atexit_handler_from_atexit() {
  *atexit_sequence += " Dumpty";
  // register 2 others
  atexit(atexit_handler_from_atexit_from_atexit2);
  atexit(atexit_handler_from_atexit_from_atexit1);
}

// 1
static void atexit_handler_with_atexit() {
  *atexit_sequence += "Humpty";
  atexit(atexit_handler_from_atexit);
}

// last
static void atexit_handler_regular() {
  *atexit_sequence += " a wall";
}

// attribute c-tor and d-tor
static void __attribute__((constructor)) atexit_attr_ctor() {
  attr_ctor_called = 1;
}

static void __attribute__((destructor)) atexit_attr_dtor() {
  if (atexit_attr_dtor_called) {
    *atexit_attr_dtor_called = true;
  }
}

extern "C" void register_atexit(std::string* sequence, bool* valid_this_in_static_dtor, bool* attr_dtor_called) {
  atexit_sequence = sequence;
  atexit_valid_this_in_static_dtor = valid_this_in_static_dtor;
  atexit_attr_dtor_called = attr_dtor_called;
  atexit(atexit_handler_regular);
  atexit(atexit_handler_with_atexit);
}

extern "C" int get_cxx_ctor_called() {
  return cxx_ctor_called;
}

extern "C" int get_attr_ctor_called() {
  return attr_ctor_called;
}