1 //===-- Unittests for atexit ----------------------------------------------===//
2 //
3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4 // See https://llvm.org/LICENSE.txt for license information.
5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6 //
7 //===----------------------------------------------------------------------===//
8
9 #include "src/__support/CPP/array.h"
10 #include "src/__support/CPP/utility.h"
11 #include "src/stdlib/atexit.h"
12 #include "src/stdlib/exit.h"
13 #include "test/UnitTest/Test.h"
14
15 static int a;
TEST(LlvmLibcAtExit,Basic)16 TEST(LlvmLibcAtExit, Basic) {
17 // In case tests ever run multiple times.
18 a = 0;
19
20 auto test = [] {
21 int status = LIBC_NAMESPACE::atexit(+[] {
22 if (a != 1)
23 __builtin_trap();
24 });
25 status |= LIBC_NAMESPACE::atexit(+[] { a++; });
26 if (status)
27 __builtin_trap();
28
29 LIBC_NAMESPACE::exit(0);
30 };
31 EXPECT_EXITS(test, 0);
32 }
33
TEST(LlvmLibcAtExit,AtExitCallsSysExit)34 TEST(LlvmLibcAtExit, AtExitCallsSysExit) {
35 auto test = [] {
36 LIBC_NAMESPACE::atexit(+[] { _Exit(1); });
37 LIBC_NAMESPACE::exit(0);
38 };
39 EXPECT_EXITS(test, 1);
40 }
41
42 static int size;
43 static LIBC_NAMESPACE::cpp::array<int, 256> arr;
44
45 template <int... Ts>
register_atexit_handlers(LIBC_NAMESPACE::cpp::integer_sequence<int,Ts...>)46 void register_atexit_handlers(
47 LIBC_NAMESPACE::cpp::integer_sequence<int, Ts...>) {
48 (LIBC_NAMESPACE::atexit(+[] { arr[size++] = Ts; }), ...);
49 }
50
getTest()51 template <int count> constexpr auto getTest() {
52 return [] {
53 LIBC_NAMESPACE::atexit(+[] {
54 if (size != count)
55 __builtin_trap();
56 for (int i = 0; i < count; i++)
57 if (arr[i] != count - 1 - i)
58 __builtin_trap();
59 });
60 register_atexit_handlers(
61 LIBC_NAMESPACE::cpp::make_integer_sequence<int, count>{});
62 LIBC_NAMESPACE::exit(0);
63 };
64 }
65
TEST(LlvmLibcAtExit,ReverseOrder)66 TEST(LlvmLibcAtExit, ReverseOrder) {
67 // In case tests ever run multiple times.
68 size = 0;
69
70 auto test = getTest<32>();
71 EXPECT_EXITS(test, 0);
72 }
73
TEST(LlvmLibcAtExit,Many)74 TEST(LlvmLibcAtExit, Many) {
75 // In case tests ever run multiple times.
76 size = 0;
77
78 auto test = getTest<256>();
79 EXPECT_EXITS(test, 0);
80 }
81
TEST(LlvmLibcAtExit,HandlerCallsAtExit)82 TEST(LlvmLibcAtExit, HandlerCallsAtExit) {
83 auto test = [] {
84 LIBC_NAMESPACE::atexit(
85 +[] { LIBC_NAMESPACE::atexit(+[] { LIBC_NAMESPACE::exit(1); }); });
86 LIBC_NAMESPACE::exit(0);
87 };
88 EXPECT_EXITS(test, 1);
89 }
90