1 /* $OpenBSD: atexit_test.c,v 1.10 2024/03/05 19:27:47 miod Exp $ */ 2 3 /* 4 * Copyright (c) 2002 Daniel Hartmeier 5 * All rights reserved. 6 * 7 * Redistribution and use in source and binary forms, with or without 8 * modification, are permitted provided that the following conditions 9 * are met: 10 * 11 * - Redistributions of source code must retain the above copyright 12 * notice, this list of conditions and the following disclaimer. 13 * - Redistributions in binary form must reproduce the above 14 * copyright notice, this list of conditions and the following 15 * disclaimer in the documentation and/or other materials provided 16 * with the distribution. 17 * 18 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 19 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 20 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS 21 * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE 22 * COPYRIGHT HOLDERS OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, 23 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, 24 * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; 25 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER 26 * CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 27 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN 28 * ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE 29 * POSSIBILITY OF SUCH DAMAGE. 30 * 31 * Effort sponsored in part by the Defense Advanced Research Projects 32 * Agency (DARPA) and Air Force Research Laboratory, Air Force 33 * Materiel Command, USAF, under agreement number F30602-01-2-0537. 34 * 35 */ 36 37 #include <stdio.h> 38 #include <stdlib.h> 39 #include <string.h> 40 #include <signal.h> 41 #include <unistd.h> 42 43 /* 44 * XXX Provide a prototype for aligned_alloc on pre-C11 compilers to prevent 45 * inclusion of hidden/stdlib.h below to cause a missing prototype error. 46 */ 47 #if (__ISO_C_VISIBLE - 0) < 2011 48 void *aligned_alloc(size_t, size_t); 49 #endif 50 51 #include "include/namespace.h" 52 #include "hidden/stdlib.h" 53 #include "stdlib/atexit.h" 54 55 void handle_first(void); 56 void handle_middle(void); 57 void handle_last(void); 58 void handle_invalid(void *); 59 void handle_cleanup(void); 60 void handle_signal(int); 61 62 static int counter; 63 64 int 65 main(int argc, char *argv[]) 66 { 67 int i; 68 69 if (argc != 2 || (strcmp(argv[1], "-valid") && 70 strcmp(argv[1], "-invalid-atexit") && 71 strcmp(argv[1], "-invalid-cleanup"))) { 72 fprintf(stderr, "%s -valid/-invalid-atexit/-invalid-cleanup\n", 73 argv[0]); 74 return (1); 75 } 76 fprintf(stderr, "main()\n"); 77 if (atexit(handle_last)) { 78 perror("atexit(handle_last) failed"); 79 return (1); 80 } 81 for (i = 0; i < 65535; ++i) { 82 if (atexit(handle_middle)) { 83 perror("atexit(handle_middle) failed"); 84 return (1); 85 } 86 } 87 if (atexit(handle_first)) { 88 perror("atexit(handle_first) failed"); 89 return (1); 90 } 91 /* this is supposed to segfault */ 92 if (!strcmp(argv[1], "-invalid-atexit")) { 93 signal(SIGSEGV, handle_signal); 94 __atexit->fns[0].fn_ptr = handle_invalid; 95 } else if (!strcmp(argv[1], "-invalid-cleanup")) { 96 struct atexit *p = __atexit; 97 98 signal(SIGSEGV, handle_signal); 99 while (p != NULL && p->next != NULL) 100 p = p->next; 101 if (p == NULL) 102 fprintf(stderr, "p == NULL, no page found\n"); 103 p->fns[0].fn_ptr = handle_invalid; 104 } 105 __atexit_register_cleanup(handle_cleanup); 106 counter = 0; 107 fprintf(stderr, "main() returns\n"); 108 return (0); 109 } 110 111 void 112 handle_first(void) 113 { 114 fprintf(stderr, "handle_first() counter == %i\n", counter); 115 } 116 117 void 118 handle_middle(void) 119 { 120 counter++; 121 } 122 123 void 124 handle_last(void) 125 { 126 fprintf(stderr, "handle_last() counter == %i\n", counter); 127 } 128 129 void 130 handle_cleanup(void) 131 { 132 fprintf(stderr, "handle_cleanup()\n"); 133 } 134 135 void 136 handle_invalid(void *arg) 137 { 138 fprintf(stderr, "handle_invalid() THIS SHOULD HAVE SEGFAULTED INSTEAD!\n"); 139 } 140 141 void 142 handle_signal(int sigraised) 143 { 144 switch (sigraised) { 145 case SIGSEGV: 146 dprintf(STDERR_FILENO, "SIGSEGV\n"); 147 exit(0); 148 default: 149 dprintf(STDERR_FILENO, "unexpected signal caught\n"); 150 exit(1); 151 } 152 } 153