1 /*- 2 * SPDX-License-Identifier: BSD-2-Clause-FreeBSD 3 * 4 * Copyright (c) 2019 The FreeBSD Foundation 5 * All rights reserved. 6 * 7 * This software was developed by Konstantin Belousov <kib@FreeBSD.org> 8 * under sponsorship from the FreeBSD Foundation. 9 * 10 * Redistribution and use in source and binary forms, with or without 11 * modification, are permitted provided that the following conditions 12 * are met: 13 * 1. Redistributions of source code must retain the above copyright 14 * notice, this list of conditions and the following disclaimer. 15 * 2. Redistributions in binary form must reproduce the above copyright 16 * notice, this list of conditions and the following disclaimer in the 17 * documentation and/or other materials provided with the distribution. 18 * 19 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND 20 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 21 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 22 * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE 23 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 24 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 25 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 26 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 27 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 28 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 29 * SUCH DAMAGE. 30 */ 31 32 #include <sys/cdefs.h> 33 __FBSDID("$FreeBSD$"); 34 35 #include <sys/types.h> 36 #include <sys/mman.h> 37 #include <rtld_malloc.h> 38 #include "thr_private.h" 39 40 int npagesizes; 41 size_t *pagesizes; 42 static size_t pagesizes_d[2]; 43 static struct umutex thr_malloc_umtx; 44 45 void 46 __thr_malloc_init(void) 47 { 48 49 if (npagesizes != 0) 50 return; 51 npagesizes = getpagesizes(pagesizes_d, nitems(pagesizes_d)); 52 if (npagesizes == -1) { 53 npagesizes = 1; 54 pagesizes_d[0] = PAGE_SIZE; 55 } 56 pagesizes = pagesizes_d; 57 _thr_umutex_init(&thr_malloc_umtx); 58 } 59 60 static void 61 thr_malloc_lock(struct pthread *curthread) 62 { 63 64 if (curthread == NULL) 65 return; 66 curthread->locklevel++; 67 _thr_umutex_lock(&thr_malloc_umtx, TID(curthread)); 68 } 69 70 static void 71 thr_malloc_unlock(struct pthread *curthread) 72 { 73 74 if (curthread == NULL) 75 return; 76 _thr_umutex_unlock(&thr_malloc_umtx, TID(curthread)); 77 curthread->locklevel--; 78 _thr_ast(curthread); 79 } 80 81 void * 82 __thr_calloc(size_t num, size_t size) 83 { 84 struct pthread *curthread; 85 void *res; 86 87 curthread = _get_curthread(); 88 thr_malloc_lock(curthread); 89 res = __crt_calloc(num, size); 90 thr_malloc_unlock(curthread); 91 return (res); 92 } 93 94 void 95 __thr_free(void *cp) 96 { 97 struct pthread *curthread; 98 99 curthread = _get_curthread(); 100 thr_malloc_lock(curthread); 101 __crt_free(cp); 102 thr_malloc_unlock(curthread); 103 } 104 105 void * 106 __thr_malloc(size_t nbytes) 107 { 108 struct pthread *curthread; 109 void *res; 110 111 curthread = _get_curthread(); 112 thr_malloc_lock(curthread); 113 res = __crt_malloc(nbytes); 114 thr_malloc_unlock(curthread); 115 return (res); 116 } 117 118 void * 119 __thr_realloc(void *cp, size_t nbytes) 120 { 121 struct pthread *curthread; 122 void *res; 123 124 curthread = _get_curthread(); 125 thr_malloc_lock(curthread); 126 res = __crt_realloc(cp, nbytes); 127 thr_malloc_unlock(curthread); 128 return (res); 129 } 130 131 void 132 __thr_malloc_prefork(struct pthread *curthread) 133 { 134 135 _thr_umutex_lock(&thr_malloc_umtx, TID(curthread)); 136 } 137 138 void 139 __thr_malloc_postfork(struct pthread *curthread) 140 { 141 142 _thr_umutex_unlock(&thr_malloc_umtx, TID(curthread)); 143 } 144