1 /* Copyright (C) 2005-2022 Free Software Foundation, Inc.
2    This file is part of the GNU C Library.
3 
4    The GNU C Library is free software; you can redistribute it and/or
5    modify it under the terms of the GNU Lesser General Public
6    License as published by the Free Software Foundation; either
7    version 2.1 of the License, or (at your option) any later version.
8 
9    The GNU C Library is distributed in the hope that it will be useful,
10    but WITHOUT ANY WARRANTY; without even the implied warranty of
11    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
12    Lesser General Public License for more details.
13 
14    You should have received a copy of the GNU Lesser General Public
15    License along with the GNU C Library; if not, see
16    <https://www.gnu.org/licenses/>.  */
17 
18 #include <errno.h>
19 #include <stdbool.h>
20 #include <stdio.h>
21 #include <stdlib.h>
22 #include <sys/wait.h>
23 #include <tst-stack-align.h>
24 #include <unistd.h>
25 
26 static int res, fds[2], result;
27 static bool test_destructors;
28 
29 extern void in_dso (int *, bool *, int *);
30 
con(void)31 static void __attribute__ ((constructor)) con (void)
32 {
33   res = TEST_STACK_ALIGN () ? -1 : 1;
34 }
35 
des(void)36 static void __attribute__ ((destructor)) des (void)
37 {
38   if (!test_destructors)
39     return;
40 
41   char c = TEST_STACK_ALIGN () ? 'B' : 'A';
42   write (fds[1], &c, 1);
43 }
44 
45 static int
do_test(void)46 do_test (void)
47 {
48   if (!res)
49     {
50       puts ("binary's constructor has not been run");
51       result = 1;
52     }
53   else if (res != 1)
54     {
55       puts ("binary's constructor has been run without sufficient alignment");
56       result = 1;
57     }
58 
59   if (TEST_STACK_ALIGN ())
60     {
61       puts ("insufficient stack alignment in do_test");
62       result = 1;
63     }
64 
65   in_dso (&result, &test_destructors, &fds[1]);
66 
67   if (pipe (fds) < 0)
68     {
69       printf ("couldn't create pipe: %m\n");
70       return 1;
71     }
72 
73   pid_t pid = fork ();
74   if (pid < 0)
75     {
76       printf ("fork failed: %m\n");
77       return 1;
78     }
79 
80   if (!pid)
81     {
82       close (fds[0]);
83       test_destructors = true;
84       exit (0);
85     }
86 
87   close (fds[1]);
88 
89   unsigned char c;
90   ssize_t len;
91   int des_seen = 0, dso_des_seen = 0;
92   while ((len = TEMP_FAILURE_RETRY (read (fds[0], &c, 1))) > 0)
93     {
94       switch (c)
95         {
96         case 'B':
97           puts ("insufficient alignment in binary's destructor");
98           result = 1;
99           /* FALLTHROUGH */
100         case 'A':
101           des_seen++;
102           break;
103         case 'D':
104           puts ("insufficient alignment in DSO destructor");
105           result = 1;
106           /* FALLTHROUGH */
107         case 'C':
108           dso_des_seen++;
109           break;
110         default:
111           printf ("unexpected character %x read from pipe", c);
112           result = 1;
113           break;
114         }
115     }
116 
117   close (fds[0]);
118 
119   if (des_seen != 1)
120     {
121       printf ("binary destructor run %d times instead of once\n", des_seen);
122       result = 1;
123     }
124 
125   if (dso_des_seen != 1)
126     {
127       printf ("DSO destructor run %d times instead of once\n", dso_des_seen);
128       result = 1;
129     }
130 
131   int status;
132   pid_t termpid;
133   termpid = TEMP_FAILURE_RETRY (waitpid (pid, &status, 0));
134   if (termpid == -1)
135     {
136       printf ("waitpid failed: %m\n");
137       result = 1;
138     }
139   else if (termpid != pid)
140     {
141       printf ("waitpid returned %ld != %ld\n",
142 	      (long int) termpid, (long int) pid);
143       result = 1;
144     }
145   else if (!WIFEXITED (status) || WEXITSTATUS (status))
146     {
147       puts ("child hasn't exited with exit status 0");
148       result = 1;
149     }
150 
151   return result;
152 }
153 
154 #include <support/test-driver.c>
155