Add #defines for magic numbers in API.
[openssl.git] / test / memleaktest.c
1 /*
2  * Copyright 2016 The OpenSSL Project Authors. All Rights Reserved.
3  *
4  * Licensed under the OpenSSL license (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
8  */
9
10 #include <stdio.h>
11 #include <string.h>
12 #include <openssl/bio.h>
13 #include <openssl/crypto.h>
14
15 int main(int argc, char **argv)
16 {
17 #ifndef OPENSSL_NO_CRYPTO_MDEBUG
18     char *p;
19     char *lost;
20     int noleak;
21
22     p = getenv("OPENSSL_DEBUG_MEMORY");
23     if (p != NULL && strcmp(p, "on") == 0)
24         CRYPTO_set_mem_debug(1);
25     CRYPTO_mem_ctrl(CRYPTO_MEM_CHECK_ON);
26
27     lost = OPENSSL_malloc(3);
28     if (lost == NULL) {
29         fprintf(stderr, "OPENSSL_malloc failed\n");
30         return 1;
31     }
32
33     if (argv[1] && strcmp(argv[1], "freeit") == 0) {
34         OPENSSL_free(lost);
35         lost = NULL;
36     }
37
38     noleak = CRYPTO_mem_leaks_fp(stderr);
39     /* If -1 return value something bad happened */
40     if (noleak == -1)
41         return 1;
42     return ((lost != NULL) ^ (noleak == 0));
43 #else
44     return 0;
45 #endif
46 }