2 * Copyright 1995-2022 The OpenSSL Project Authors. All Rights Reserved.
4 * Licensed under the Apache License 2.0 (the "License"). You may not use
5 * this file except in compliance with the License. You can obtain a copy
6 * in the file LICENSE in the source distribution or at
7 * https://www.openssl.org/source/license.html
10 #include "internal/e_os.h"
11 #include "internal/cryptlib.h"
12 #include "crypto/cryptlib.h"
16 #include <openssl/crypto.h>
19 * the following pointers may be changed as long as 'allow_customize' is set
21 static int allow_customize
= 1;
22 static CRYPTO_malloc_fn malloc_impl
= CRYPTO_malloc
;
23 static CRYPTO_realloc_fn realloc_impl
= CRYPTO_realloc
;
24 static CRYPTO_free_fn free_impl
= CRYPTO_free
;
26 #if !defined(OPENSSL_NO_CRYPTO_MDEBUG) && !defined(FIPS_MODULE)
27 # include "internal/tsan_assist.h"
29 # ifdef TSAN_REQUIRES_LOCKING
30 # define INCREMENT(x) /* empty */
32 # else /* TSAN_REQUIRES_LOCKING */
33 static TSAN_QUALIFIER
int malloc_count
;
34 static TSAN_QUALIFIER
int realloc_count
;
35 static TSAN_QUALIFIER
int free_count
;
37 # define INCREMENT(x) tsan_counter(&(x))
38 # define LOAD(x) tsan_load(&x)
39 # endif /* TSAN_REQUIRES_LOCKING */
41 static char *md_failstring
;
43 static int md_fail_percent
= 0;
44 static int md_tracefd
= -1;
46 static void parseit(void);
47 static int shouldfail(void);
49 # define FAILTEST() if (shouldfail()) return NULL
53 # define INCREMENT(x) /* empty */
54 # define FAILTEST() /* empty */
57 int CRYPTO_set_mem_functions(CRYPTO_malloc_fn malloc_fn
,
58 CRYPTO_realloc_fn realloc_fn
,
59 CRYPTO_free_fn free_fn
)
63 if (malloc_fn
!= NULL
)
64 malloc_impl
= malloc_fn
;
65 if (realloc_fn
!= NULL
)
66 realloc_impl
= realloc_fn
;
72 void CRYPTO_get_mem_functions(CRYPTO_malloc_fn
*malloc_fn
,
73 CRYPTO_realloc_fn
*realloc_fn
,
74 CRYPTO_free_fn
*free_fn
)
76 if (malloc_fn
!= NULL
)
77 *malloc_fn
= malloc_impl
;
78 if (realloc_fn
!= NULL
)
79 *realloc_fn
= realloc_impl
;
84 #if !defined(OPENSSL_NO_CRYPTO_MDEBUG) && !defined(FIPS_MODULE)
85 void CRYPTO_get_alloc_counts(int *mcount
, int *rcount
, int *fcount
)
88 *mcount
= LOAD(malloc_count
);
90 *rcount
= LOAD(realloc_count
);
92 *fcount
= LOAD(free_count
);
96 * Parse a "malloc failure spec" string. This likes like a set of fields
97 * separated by semicolons. Each field has a count and an optional failure
98 * percentage. For example:
101 * This means 100 mallocs succeed, then next 100 fail 25% of the time, and
102 * all remaining (count is zero) succeed.
104 static void parseit(void)
106 char *semi
= strchr(md_failstring
, ';');
112 /* Get the count (atol will stop at the @ if there), and percentage */
113 md_count
= atol(md_failstring
);
114 atsign
= strchr(md_failstring
, '@');
115 md_fail_percent
= atsign
== NULL
? 0 : atoi(atsign
+ 1);
118 md_failstring
= semi
;
122 * Windows doesn't have random(), but it has rand()
123 * Some rand() implementations aren't good, but we're not
124 * dealing with secure randomness here.
127 # define random() rand()
130 * See if the current malloc should fail.
132 static int shouldfail(void)
134 int roll
= (int)(random() % 100);
135 int shoulditfail
= roll
< md_fail_percent
;
137 /* suppressed on Windows as POSIX-like file descriptors are non-inheritable */
141 if (md_tracefd
> 0) {
142 BIO_snprintf(buff
, sizeof(buff
),
143 "%c C%ld %%%d R%d\n",
144 shoulditfail
? '-' : '+', md_count
, md_fail_percent
, roll
);
146 if (write(md_tracefd
, buff
, len
) != len
)
147 perror("shouldfail write failed");
152 /* If we used up this one, go to the next. */
160 void ossl_malloc_setup_failures(void)
162 const char *cp
= getenv("OPENSSL_MALLOC_FAILURES");
164 if (cp
!= NULL
&& (md_failstring
= strdup(cp
)) != NULL
)
166 if ((cp
= getenv("OPENSSL_MALLOC_FD")) != NULL
)
167 md_tracefd
= atoi(cp
);
171 void *CRYPTO_malloc(size_t num
, const char *file
, int line
)
173 INCREMENT(malloc_count
);
174 if (malloc_impl
!= CRYPTO_malloc
)
175 return malloc_impl(num
, file
, line
);
181 if (allow_customize
) {
183 * Disallow customization after the first allocation. We only set this
184 * if necessary to avoid a store to the same cache line on every
193 void *CRYPTO_zalloc(size_t num
, const char *file
, int line
)
197 ret
= CRYPTO_malloc(num
, file
, line
);
205 void *CRYPTO_realloc(void *str
, size_t num
, const char *file
, int line
)
207 INCREMENT(realloc_count
);
208 if (realloc_impl
!= CRYPTO_realloc
)
209 return realloc_impl(str
, num
, file
, line
);
213 return CRYPTO_malloc(num
, file
, line
);
216 CRYPTO_free(str
, file
, line
);
220 return realloc(str
, num
);
223 void *CRYPTO_clear_realloc(void *str
, size_t old_len
, size_t num
,
224 const char *file
, int line
)
229 return CRYPTO_malloc(num
, file
, line
);
232 CRYPTO_clear_free(str
, old_len
, file
, line
);
236 /* Can't shrink the buffer since memcpy below copies |old_len| bytes. */
238 OPENSSL_cleanse((char*)str
+ num
, old_len
- num
);
242 ret
= CRYPTO_malloc(num
, file
, line
);
244 memcpy(ret
, str
, old_len
);
245 CRYPTO_clear_free(str
, old_len
, file
, line
);
250 void CRYPTO_free(void *str
, const char *file
, int line
)
252 INCREMENT(free_count
);
253 if (free_impl
!= CRYPTO_free
) {
254 free_impl(str
, file
, line
);
261 void CRYPTO_clear_free(void *str
, size_t num
, const char *file
, int line
)
266 OPENSSL_cleanse(str
, num
);
267 CRYPTO_free(str
, file
, line
);
270 #if !defined(OPENSSL_NO_CRYPTO_MDEBUG)
272 # ifndef OPENSSL_NO_DEPRECATED_3_0
273 int CRYPTO_mem_ctrl(int mode
)
279 int CRYPTO_set_mem_debug(int flag
)
285 int CRYPTO_mem_debug_push(const char *info
, const char *file
, int line
)
287 (void)info
; (void)file
; (void)line
;
291 int CRYPTO_mem_debug_pop(void)
296 void CRYPTO_mem_debug_malloc(void *addr
, size_t num
, int flag
,
297 const char *file
, int line
)
299 (void)addr
; (void)num
; (void)flag
; (void)file
; (void)line
;
302 void CRYPTO_mem_debug_realloc(void *addr1
, void *addr2
, size_t num
, int flag
,
303 const char *file
, int line
)
305 (void)addr1
; (void)addr2
; (void)num
; (void)flag
; (void)file
; (void)line
;
308 void CRYPTO_mem_debug_free(void *addr
, int flag
,
309 const char *file
, int line
)
311 (void)addr
; (void)flag
; (void)file
; (void)line
;
314 int CRYPTO_mem_leaks(BIO
*b
)
320 # ifndef OPENSSL_NO_STDIO
321 int CRYPTO_mem_leaks_fp(FILE *fp
)
328 int CRYPTO_mem_leaks_cb(int (*cb
)(const char *str
, size_t len
, void *u
),