xref: /freebsd/lib/libthr/thread/thr_malloc.c (revision 2f513db72b034fd5ef7f080b11be5c711c15186a)
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